树状数组求逆序对

4658. 【GDOI2017模拟7.19】小X调顺序 (Standard IO)
      Time Limits: 1000 ms Memory Limits: 262144 KB Detailed Limits

Description

Input

Output

Sample Input
3 1
2 2 1

Sample Output
1

Data Constraint

Formula
30 pts

#include<bits/stdc++.h>
using namespace std;
int n,k,i,j,a[1111111],nsd; 
int main()
{
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	nsd=0;
	for(i=1;i<=n-1;i++)
	{
		for(j=i+1;j<=n;j++)
		{
			if(a[i]>a[j]) nsd++;
		}
	}
	nsd-=k;
	if(nsd<0) nsd=0;
	printf("%d\n",nsd);
}

100pts
树状数组

#include<bits/stdc++.h>
using namespace std;
long long n,k,i,j,a[1111111],nsd,d[1111111],t[1111111];
long long lb(long long xx)
{
	return -xx&xx;
}
void add(long long xx,long long yy)
{
	while(xx<=n)
	{
		t[xx]+=yy;
		xx+=lb(xx);
	}
}
long long sum(long long xx)
{
	long long ans=0;
	while(xx>=1)
	{
		ans+=t[xx];
		xx-=lb(xx);
	}
	return ans;
}
bool cmp(long long xx,long long yy)
{
	if(a[xx]==a[yy]) return xx>yy;
	else return a[xx]>a[yy];
}
void kp(long long l,long long r)
{
	long long i=l,j=r,mid=a[(l+r)/2];
	while(i<=j)
	{
		while(a[i]>mid) i++;
		while(a[j]<mid) j--;
		if(i<=j)
		{
			a[0]=a[i];
			a[i]=a[j];
			a[j]=a[0];
			d[0]=d[i];
			d[i]=d[j];
			d[j]=d[0];
			i++;
			j--;
		}
	}
	if(i<r) kp(i,r);
	if(l<j) kp(l,j);
}
int main()
{
	scanf("%lld%lld",&n,&k);
	for(i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		d[i]=i;
	}
	sort(d+1,d+n+1,cmp);
	nsd=0;
	for(i=1;i<=n;i++)
	{
		add(d[i],1);
		nsd+=(sum(d[i]-1));
	}
	nsd-=k;
	if(nsd<0) nsd=0;
	printf("%lld\n",nsd);
}