码迷,mamicode.com
首页 > 其他好文 > 详细

nyist oj 117 求逆序数 (归并排序&&树状数组)

时间:2014-10-01 19:33:21      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:acm   算法   归并排序   

求逆序数

时间限制:2000 ms  |  内存限制:65535 KB
难度:5
描述

在一个排列中,如果一对数的前后位置与大小顺序相反,即前面的数大于后面的数,那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。

现在,给你一个N个元素的序列,请你判断出它的逆序数是多少。

比如 1 3 2 的逆序数就是1。

输入
第一行输入一个整数T表示测试数据的组数(1<=T<=5)
每组测试数据的每一行是一个整数N表示数列中共有N个元素(2〈=N〈=1000000)
随后的一行共有N个整数Ai(0<=Ai<1000000000),表示数列中的所有元素。

数据保证在多组测试数据中,多于10万个数的测试数据最多只有一组。
输出
输出该数列的逆序数
样例输入
2
2
1 1
3
1 3 2
样例输出
0
1
来源
[张云聪]原创
上传者
张云聪

求逆序数:用归并排序来做,效率还是很高的;
归并排序利用分治的思想,先把一个数组分成一个个序列,然后对一个个序列排序,把排好序的序列,在合并到原来的数组中。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000001;
int a[maxn],b[maxn];
long long sum;
void Merge(int begin,int mid,int end)//归并
{
    int i=begin,j=mid+1,pos=begin;
    while(i<=mid && j<=end)//对一个个序列排序的过程
    {
        if(a[i]<=a[j])
        {
            b[pos++]=a[i++];
        }
        else
        {
            b[pos++]=a[j++];
            sum+=mid-i+1;//求逆序数
        }
    }
    while(i<=mid) b[pos++]=a[i++];
    while(j<=end) b[pos++]=a[j++];
    for(int i=begin,j=begin;i<=end;i++,j++)
        a[i]=b[j];
}
void Sort(int begin,int end)//排序
{
    if(begin<end)
    {
        int mid=(begin+end)/2;
        Sort(begin,mid);
        Sort(mid+1,end);
        Merge(begin,mid,end);
    }
}
int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
        sum=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
            Sort(1,n);
        printf("%lld\n",sum);
    }
    return 0;
}

用树状数组求逆序数:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000005;
struct node
{
   int val,id;
}s[maxn];
int a[maxn],n;
int lowbit(int i)//位运算
{
    return i&(-i);
}
void update(int i)//更新
{
    while(i<=n)
    {
        a[i]++;
        i+=lowbit(i);
    }
}
int sum(int i)//求和
{
    int sum=0;
    while(i>0)
    {
        sum+=a[i];
        i-=lowbit(i);
    }
    return sum;
}
bool cmp(node x,node y)//比较函数
{
    if(x.val!=y.val)
        return x.val<y.val;
    return x.id<y.id;
}
int main()
{
    int t,i;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&s[i].val);
            s[i].id=i;
            a[i]=0;
        }
        long long ans=0;
        sort(s+1,s+n+1,cmp);//离散化
        for(i=1;i<=n;i++)
        {
            update(s[i].id);
            ans+=i-sum(s[i].id);
        }
        printf("%lld\n",ans);
    }
    return 0;
}



nyist oj 117 求逆序数 (归并排序&&树状数组)

标签:acm   算法   归并排序   

原文地址:http://blog.csdn.net/whjkm/article/details/39718765

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!