hust1433-樹狀陣列-2
阿新 • • 發佈:2018-11-01
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; }