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;}