Tyvj P1384 精灵魔法

精灵魔法

一道果的求逆序对的问题

思路

就是很裸的归并求逆序对(也可以用树状数组)

在归并排序的过程中交换了几次就有几个逆序对(结果用long long)

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mid (l+r)/2
using namespace std;

const int Maxn=100001;

#define LL long long

struct node {LL pos,v;}a[Maxn];

bool comp(node a,node b){return a.pos<b.pos;}

LL n,ans,temp[Maxn];

void merge(int l,int r)
{
	int i=l,j=mid+1,k=l;
	while(i<=mid&&j<=r){
		if(a[i].v>a[j].v){
			temp[k++]=a[j++].v;
			ans += mid-i+1;	
		}
		else temp[k++]=a[i++].v;
	}
	while(i<=mid)temp[k++]=a[i++].v;
	while(j<=r)temp[k++]=a[j++].v;
	for(int i=l;i<=r;i++)a[i].v=temp[i];
}

void merge_sort(int l,int r)
{
	if(l<r){
		merge_sort(l,mid);
		merge_sort(mid+1,r);
		merge(l,r);
	}
}

int main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i].pos);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i].v);
	sort(a+1,a+n+1,comp);
	
	merge_sort(1,n);
	
	printf("%lld",ans);
}

欢迎指出Bug

End。