UVA 1513 - Movie collection（树状数组）

标签：

题目链接：点击打开链接

题意： 有编号1~n的n个影碟从上到下排列， 每次取一个影碟并把其放在最上面， 求每次取之前该影碟前面有多少个影碟。

取出影碟， 将该位置-1即可， 容易想到用树状数组来维护， 但是还要放到最前面。 其实解决方法很简单， 就是把数组开大一点， 前面留出足够大的空间， 不断更新位置即可。

细节参见代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int mod = 1000000000 + 7;
const int INF = 1000000000;
const int maxn = 200000 + 10;
int T,n,v,m,bit[maxn],pos[maxn>>1];
int sum(int x) {
    int ans = 0;
    while(x > 0) {
        ans += bit[x];
        x -= x & -x;
    }
    return ans;
}
void add(int x, int d) {
    while(x <= n + m) {
        bit[x] += d;
        x += x & -x;
    }
}
int main() {
    scanf("%d",&T);
    while(T--) {
        scanf("%d%d",&n,&m);
        memset(bit, 0, (n+m+1)*sizeof(bit[0]));
        int cnt = m;
        for(int i=1;i<=n;i++) {
            pos[i] = i+m;
            add(pos[i], 1);
        }
        for(int i=0;i<m;i++) {
            scanf("%d",&v);
            printf("%d%c", sum(pos[v]-1), i == m-1 ? '\n' : ' ');
            add(pos[v], -1);
            pos[v] = cnt--;
            add(pos[v], 1);
        }
    }
    return 0;
}

UVA 1513 - Movie collection（树状数组）

标签：

原文地址：http://blog.csdn.net/weizhuwyzc000/article/details/50542459