Movie collection UVA - 1513（樹狀數組模板題）

學習時參考了這篇文章，看完基本就能弄懂樹狀數組是怎麼回事了：http://blog.csdn.net/int64ago/article/details/7429868

題目鏈接：https://vjudge.net/contest/180021#problem/D

題目描述：一開始共有編號1~n共n張影片從上之下排列，每次抽出一張影片x，求抽出時影片x上面有幾張影片，每次將抽出的影片放在最上面。

思路：搜了題解說是樹狀數組，然後就去網上學，說真的，真的不怪別人瞧不起自己是我自己能力不高，之前一年多時間沒把心思主要放在ACM上，連樹狀數組模板題都寫不出來，現在除了讓自己變強，別的都是浮雲，什麼都不想考慮但求問心無愧。一不小心就跑題了，還是趕緊回來做正事吧。每次抽出一張影片x放到最上面，它原來位置下面的影片不會受影響，上面的影片在它們上面的影片數會加1（因爲x放到了它們最上面），所以考慮將影片逆序排列，即最下面的位置是1，最上面是n，這樣就可以用樹狀數組來維護。

代碼如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<cstdlib>
#include<sstream>
#include<deque>
#include<stack>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps = 1e-6;
const int  maxn = 200000 + 10;
const int  maxt = 300 + 10;
const int mod = 10;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, -1, 1};
const int Dis[] = {-1, 1, -5, 5};
const int inf = 0x3f3f3f3f;
const int MOD = 1000;
int n, m, k;
int tree[maxn], pos[maxn];
inline int lowbit(int x){//求x只留下最低位的1後所得數字，如：lowbit(100110) = 10, lowbit(1000) = 1000
    return x & -x;
}
void update(int x, int num){//x位置的值增加num，維護新的樹狀數組值
    while(x < maxn){//修改的是x及其之後整個區間
        tree[x] += num;
        x += lowbit(x);
    }
}
int getsum(int x){//求區間[1,x]的區間和
    int sum = 0;
    while(x >= 1){
        sum += tree[x];
        x -= lowbit(x);
    }
    return sum;
}
int main(){
    int t;
    scanf("%d", &t);
    while(t--){
        memset(tree, 0, sizeof tree);
        scanf("%d%d", &n, &m);
        for(int i = 1; i <= n; ++i){
            pos[i] = n - i + 1;//1~n倒序排列
            update(i, 1);
        }
        bool ok = false;
        int x;
        int cnt = n;
        for(int i = 1; i <= m; ++i){
            scanf("%d", &x);
            if(ok) printf(" ");
            else ok = true;
            printf("%d", n - getsum(pos[x]));//x之上有多少影片
            update(pos[x], -1);//要將x抽出，故x上面的所有位置的值減一，因爲他們下面少了一張影片
            pos[x] = ++cnt;//x跑到最上面，倒序排列即tree數組最後面
            update(pos[x], 1);//更新最後面的位置值
        }
        printf("\n");
    }
    return 0;
}