UVa 1513 Movie collection （树状数组）

题目链接：http://acm.hust.edu.cn/vjudge/problem/36169

题意：有n个数按1,2...,n从上到下排列，每次选择一个数，输出这个数上面有几个数，选择后的数放在最顶端。

思路：

在这个题中，每个数都有自己的位置，我们让位置编号从上到下依次递减，用数组C[i]表示第i个位置有几个数（只能是0或1），那么sum（i）就表示第1到第i个位置一共有几个数，即i以及它的下面一共有几个数，用n减去sum（i）即为答案。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lc rt<<1
#define rc rt<<1|1
using namespace std;

typedef long long ll;
const int maxn = 200000 + 10;
const int INF = 1e9 + 10;
int pos[maxn],C[maxn];
vector<int> ans;
int m,n;

int lowbit(int x)
{
return x & -x;
}

int sum(int x)
{
int cnt = 0;
while(x > 0) {
cnt += C[x];
x -= lowbit(x);
}
return cnt;
}

int add(int x, int p)
{
while(x < maxn) {
C[x] += p;
x += lowbit(x);
}
}

int main()
{
int T;
scanf("%d",&T);
while(T--) {
memset(C, 0, sizeof C);
ans.clear();
scanf("%d%d",&n,&m);

for(int i = 1; i <= n; i++) {
pos[i] = n-i+1;
add(pos[i],1);
}

int cnt = n;
while(m--) {
int x;
scanf("%d",&x);
ans.push_back(n-sum(pos[x]));
add(pos[x],-1);
pos[x] = ++cnt;
add(pos[x],1);
}

for(int i = 0; i < ans.size(); i++) {
if (i) printf(" ");
printf("%d",ans[i]);
}
printf("\n");
}
}