Minimum Inversion Number 数状数组

 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <algorithm>
 4 using namespace std;
 5 
 6 int n,c[5050];
 7 
 8 int lowbit(int x)
 9 {
10     return x&(-x);
11 }
12 
13 void add(int i,int val)
14 {
15     while(i<=n)
16     {
17         c[i]=c[i]+val;
18         i=i+lowbit(i);
19     }
20 }
21 
22 int sum(int i)
23 {
24     int s=0;
25     while(i)
26     {
27         s=s+c[i];
28         i=i-lowbit(i);
29     }
30     return s;
31 }
32 
33 int main()
34 {
35     int i,j,k;
36     int a[5050];
37     while(scanf("%d",&n)!=EOF)
38     {
39         int cnt=0;
40         memset(c,0,sizeof(c));
41         for(i=1;i<=n;i++)
42         {
43             scanf("%d",&a[i]);
44             a[i]++;
45             cnt=cnt+sum(n)-sum(a[i]);
46             add(a[i],1);
47         }
48         int mi=cnt;
49         for(i=1;i<n;i++)
50         {
51             cnt=cnt-(a[i]-1)+(n-a[i]);
52             if(cnt<mi)
53                 mi=cnt;
54         }
55         printf("%d\n",mi);
56     }
57     return 0;
58 }