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[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;
}