暑假集训之树状数组C题（C - Minimum Inversion Number）

一道经典的逆序对的题目：

要求一个长度为n的序列的n种不同循环同构体的最小逆序对数；
n<=5000,由于多组测试样例暴力肯定肯定不行，不过我还是试了三发，都tle了；

先上暴力的错误代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#define ll long long  
using namespace std;
const int N=5010;
int n;

int t[N];
int a[N];
int cnt=0;
int lowbit(int x){
	return x&(-x);
}
void add(int x,int k){
	for(int i=x;i<=n;i+=lowbit(i))t[i]+=k;
}
int ask(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=t[i];
	
	return res;
}
int main(){
  	while(scanf("%d",&n)!=EOF){
	int res=(1<<30);
	for(int i=0;i<n;++i){
		scanf("%d",&a[i]);
		++a[i];
	}
	 for(int i=0;i<n;++i)
	{
		int sum=0;
		memset(t,0,sizeof t);
		for(int j=0;j<n;++j)
		{
			int tt=a[(i+j)%n];
			add(tt,1);
			sum+=(ask(n)-ask(tt));
		}
		res=min(sum,res);
 	}    
    printf("%d\n",res);
  	}
    return 0;
    
}

思路很简单：
再次审题的时候发现序列是0~n-1的一个全排列；
就可以联想当首元素移到队尾时逆序对的和的变化是有规律的

比如，a[i]移到队尾，则减少的逆序对数量就是a[i]-1;
增加的逆序对数量是n-a[i]
则res的变化是
res+=-(a[i]-1)+n-a[i];
res+=n-2*a[i]+1;
故AC:

#include<iostream>
#include<algorithm>
#include<cstring>
#define ll long long  
using namespace std;
const int N=5010;
int n;

int t[N];
int a[N];
int ans[N];
int lowbit(int x){
	return x&(-x);
}
void add(int x,int k){
	for(int i=x;i<=n;i+=lowbit(i))t[i]+=k;
}
int ask(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=t[i];
	
	return res;
}
int main(){
  	while(scanf("%d",&n)!=EOF){
  	memset(ans,0,sizeof ans);
  	memset(t,0,sizeof t);
  	memset(a,0,sizeof a);
	int res=0;
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i]);
		++a[i];
		add(a[i],1);
		ans[i]=(i)-ask(a[i]);
		res+=ans[i];
	}
	int sum=res;
	 for(int i=1;i<=n;++i)
	{
		sum+=n-2*a[i]+1;
		
		res=min(sum,res);
 	}    
    printf("%d\n",res);
  	}
    return 0;
    
}