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