这算是一道比较综合的树状数组题。。
题意:一个农场主养了很多奶牛,每天晚上该农场主都要为奶牛,但是每个奶牛都有一个脾气,这可能会导致奶牛损坏农场主喂牛的工具。。每个奶牛的脾气不等,这样农场主可以调换的某两个牛的位置,以求奶牛破坏最少的工具。已知挪动两个奶牛花费的时间为两个奶牛脾气的和。。让你求出最少的时间在破坏最少工具的前提下。。
思路:树状数组中有两个元素一个是记录比当前a小的个数,一个是记录比当前a小的数的和。。mintime=a*(i-Quary_cnt(a))+Quary_sum(n)-Quary_sum(a);
#include<iostream> #include<cstdio> #include<string.h> #include<algorithm> #define M 100001 using namespace std; typedef long long L; int n; typedef struct str { int cnt; L sum; }Node; Node s[M]; inline int lowbit(int x) {return x&(-x);} void update(int x,int y) { while(x<=n) { s[x].sum+=y; s[x].cnt++; x+=lowbit(x); } } L Quary_sum(int x) { L res=0; while(x>0) { res+=s[x].sum; x-=lowbit(x); } return res; } L Quary_cnt(int x) { L res=0; while(x>0) { res+=s[x].cnt; x-=lowbit(x); } return res; } int main() { while(~scanf("%d",&n)) { memset(s,0,sizeof(s)); int a; L s1=0; for(int i=1;i<=n;++i) { scanf("%d",&a); update(a,a); L res=i-Quary_cnt(a);//求逆序数的个数 if(res!=0) { L k=Quary_sum(n)-Quary_sum(a);//比a大的所有元素的和 s1+=(k+res*a); } } printf("%I64d\n",s1); }return 0; }