POJ 1084 Brainman(分治,归并排序求逆序数)

求数列的逆序数,用分治的方式,在归并排序的过程中顺便求逆序数,在排序时逆序数的可分解和可传递性与排序很像。当数组两边的顺序都排好后,可在O(n)下计算一个数在左边,一个数在右边的逆序数对数。比如,

6 8 7 9 0 3 2 1

在最后一次merge时,左边为 6 7 8 9 右边为 0 1 2 3, 然后分别对两个数按照归并排序合并的顺序比较,假设两边的数组分别为a和b,由于两边都是升序的,当左边的a[i]>b[j]时,可以知道对于所有的k>=i,a[k]>b[j]成立,比如a[0]=6>b[0]=0=>左边的a[0]一直到a[3]都大于b[0],此时将答案增加size(a)-i+1 并让j++即可, 也就是代码中的ans+=mid-p1+1,tmp[p++]=a[p2++]

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<sstream>
#include<cmath>
#include<iterator>
#include<bitset>
#include<stdio.h>
#include<time.h>
using namespace std;
#define _for(i,a,b) for(int i=(a);i<(b);++i)
#define _rep(i,a,b) for(int i=(a);i<=(b);++i)
typedef long long LL;
const int INF = 0xffffff0;
const int MOD = 1e9 + 7;
const int maxn = 1005;

int mergesort(int a[], int L,int R,int *tmp){
    if(L>=R)return 0;
    int ans=0;
    int mid=L+(R-L)/2;
    ans+=mergesort(a,L,mid,tmp);
    ans+=mergesort(a,mid+1,R,tmp);

    int p1=L,p2=mid+1,p=0;
    while(p1<=mid&&p2<=R){
        if(a[p1]>a[p2]) {
            ans+=mid-p1+1;
            tmp[p++]=a[p2++];
        }
        else tmp[p++]=a[p1++];
    }
    while(p1<=mid) {
        tmp[p++]=a[p1++];
    }
    while(p2<=R)   tmp[p++]=a[p2++];
    p=0;
    for(int i=L;i<=R;++i)a[i]=tmp[p++];
    return ans;
}

int main() {
	//freopen("C:\\Users\\admin\\Desktop\\in.txt", "r", stdin);
	//freopen("C:\\Users\\admin\\Desktop\\out.txt", "w", stdout);
    int T;scanf("%d",&T);
    int n;int a[maxn],tmp[maxn];
    int kase=0;
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<n;++i)scanf("%d",&a[i]);
        printf("Scenario #%d:\n%d\n\n",++kase,mergesort(a,0,n-1,tmp));
    }

	return 0;
}

猜你喜欢

转载自blog.csdn.net/tomandjake_/article/details/81490267