hust1433-树状数组-2

http://acm.hust.edu.cn/JudgeOnline/problem.php?id=1433

题意：给定一个1...n的排列，对i<=k<j. ans[k]为a[i]>a[j]的对数，求ans[1...n-1]。。。

分析：用树状数组，求出k位置前面>=a[k]的和个数ll[k]后面<a[k]的个数rr[k]，ans[k] = ans[k-1] + rr[k] - ll[k]+1

还有大牛直接用一个公式做的：ans[k] = ans[k-1] + a[i] - i;....直接做啊。。表示不怎么懂。。。

代码：

#include <stdio.h>
#include <iostream>
using namespace std;

#define  __int64 long long

const int N=100001;
int a, n, f
;
__int64 ans
, ll
, rr
, tmp;

void insert(int i)
{
for(; i<=n; i+=i&(-i))
f[i] += 1;
}
int query(int i)
{
int tmp=0;
for(; i>0; i-=i&(-i))
tmp += f[i];
return tmp;
}

int main()
{
int i, cas, cas1=1;
while(scanf("%d", &n)!=EOF)
{
for(i=1; i<=n; i++)
f[i] = 0;
ans[0] = 0;
for(i=1; i<=n; i++)
{
scanf("%d", &a);
insert(a);
tmp = query(a-1); //前面小于a[i]的
ll[i] = i - tmp; //前面>=a[i]的
rr[i] = a-1-tmp; //后面<a[i]的
ans[i] = ans[i-1] + rr[i] - (ll[i]-1);
}
for(i=1; i<n-1; i++)
printf("%lld ", ans[i]);
printf("%lld\n", ans[i]);
}

return 0;
}