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[N];
__int64 ans[N], ll[N], rr[N], 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;
}