uva 1513 - Movie collection(树状数组)

题目连接：uva 1513 - Movie collection

题目大意：一个人喜欢收集电影，电影的碟子从小到大放好，现在给定他每次要看的电影编号，输出有多少个碟子在该电影的上端，每次看完一个将会放回顶端。

解题思路：树状数组，用一个数组记录每个编号的电影现在所处的位置。然后用树状数组维护个数即可。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
#define lowbit(x) ((x)&(-x))
const int maxn = 1e5+5;

int N, M, fenw[maxn*2], pos[maxn];

void add (int x, int v) {
    while (x < 2*maxn) {
        fenw[x] += v;    
        x += lowbit(x);
    }
}

int sum (int x) {
    int ret = 0;
    while (x) {
        ret += fenw[x];
        x -= lowbit(x);
    }
    return ret;
}

int main () {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        scanf("%d%d", &N, &M);
        memset(fenw, 0, sizeof(fenw));
        int n = N, x;
        for (int i = 1; i <= N; i++) {
            pos[i] = N - i + 1;
            add(pos[i], 1);
        }

        for (int i = 1; i <= M; i++) {
            scanf("%d", &x);
            printf("%d%c", n - sum(pos[x]), i == M ? '\n' : ' ');
            add(pos[x], -1);
            pos[x] = ++N;
            add(pos[x], 1);
        }
    }
    return 0;
}