POJ 3270 / HDU 2838 - Cow Sorting

 

题目地址：  http://poj.org/problem?id=3270

 

       注意到了其中一部分需要__int64，另一部分被想当然的忽略了（以为不可能爆int）。结果WA了n久不知原因。

 

       很经典的求逆序数题目。初学题。比基本的逆序数多了一个状态。即需要多写一个Query。

 

 

#include<iostream>
#include<cstdio>

using namespace std;

__int64 tree[100100];
__int64 num[100100];

int n;

__int64 s[100100];

__int64 query(__int64 val){
	__int64 sum=0;
	while(val>0){
		sum+=tree[val];
		val-=val&-val;
	}
	return sum;
}
__int64 query2(__int64 val){
	__int64 sum=0;
	while(val>0){
		sum+=num[val];
		val-=val&-val;
	}
	return sum;
}
void update(int val){
	__int64 temp=val;
	while(val<=n){
		tree[val]+=temp;
		num[val]++;
		val+=val&-val;
	}
}

int main(){
	int i;
	__int64 sum=0;
	while(~scanf("%d",&n)){
		sum=0;
		for(i=1;i<=n;i++)
			tree[i]=num[i]=0;
		for(i=1;i<=n;i++)
			scanf("%I64d",&s[i]);
		for(i=n;i>=1;i--){
			sum+=query(s[i]-1)+query2(s[i]-1)*s[i];
	//		printf("%d\n",s[i]);
			update(s[i]);
	//		cout<<"tree"<<": ";for(int j=1;j<=n;j++)printf("%3I64d",tree[j]);cout<<endl;
	//		cout<<"num "<<": ";for(int j=1;j<=n;j++)printf("%3d",num[j]);cout<<endl;
		}
		printf("%I64d\n",sum);
	}
	return 0;
}