hdu2838

http://acm.hdu.edu.cn/showproblem.php?pid=2838

思路：对于数据3，4，1，2

先按顺序建树，updata(a[i],a[i]),s+=a[i];
当i=3时，1前面有3，4比1大，这时，先求出比1等于或小的元素和，在用s-sum(1),结果为，1前面比1大的元素和。

代码：

View Code 
 1 #include<iostream>
 2 using namespace std;
 3 __int64 a[100005],c[100005],d[100005],n;
 4 int lowbit(int x)
 5 {
 6     return x&(-x);
 7 }
 8 __int64 sumc(int x)
 9 {
10     __int64 s=0;
11     while(x>0)
12     {
13         s+=c[x];
14         x-=lowbit(x);
15     }
16     return s;
17 }
18 __int64 sumd(int x)
19 {
20     __int64 s=0;
21     while(x>0)
22     {
23         s+=d[x];
24         x-=lowbit(x);
25     }
26     return s;
27 }
28 void updata(int i,int j)
29 {
30     while(i<=n)
31     {
32         c[i]+=1;
33         d[i]+=j;
34         i+=lowbit(i);
35     }
36 }
37 int main()
38 {
39 
40     while(scanf("%I64d",&n)>0)
41     {
42         memset(a,0,sizeof(a));
43         memset(c,0,sizeof(c));
44         memset(d,0,sizeof(d));
45         __int64 s=0,count=0;
46         for(int i=1;i<=n;i++)
47         {
48             scanf("%d",&a[i]);
49             updata(a[i],a[i]);
50             count+=a[i];               //a[1]~~a[i]之和
51             s+=(i-sumc(a[i]))*a[i];    //a[i]与前面比a[i]大的数交换的次数*a[i]
52             s+=count-sumd(a[i]);     //a[1]~~a[i]的总和-a[i]前面比a[i]小的数之和
53         }
54         printf("%I64d\n",s);
55     }
56     return 0;
57 }

 

posted @ 2012-11-22 15:48  紫忆  阅读(238)  评论(0)  编辑  收藏  举报