【UVA】1513-Movie collection（树状数组）

树状数组是还是做的很吃力，这道题大意就是一道栈堆的模拟，不过需要稍微思考一下对应的方法。

我们可以假设栈堆的大小为 n + m，只不过一开始前 m 是空的而已，我们从第m + 1个位置导入元素到栈堆，那么我们可以利用x[i]表示位置i的地方是否存在一个元素，如果存在的话，x[i] = 1，如果不存在的话 x[i] = 0，因此问题就转化成了 给出一个数的位置pos，求x[1] + x[2] + ……+ x[pos - 1]的问题，然后每次我们把一个元素从pos位置处取出来，将他放在pos2的位置处，我们需要修改 x[pos2] + x[pos2 + 1]+ ……+x[pos] 的值。此外利用vis[i]记录i的位置，每次取出i之后更新i的位置。

思路还是不够，需要锻炼啊~

13965806

1513

Movie collection

Accepted

C++

0.189

2014-07-30 06:42:44

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define MAXD 200000 + 10
int vis[MAXD];  /*这个数组用来记录i的位置*/
int C[MAXD];  /*这个就是树状数组了*/
int n,m;
int lowbit(int x){
    return x & -x;
}
int Sum(int x){
    int ret = 0;
    while(x > 0){
        ret += C[x];
          x -=lowbit(x);
    }
    return ret;
}
void Add(int x,int d){
    while(x <= n + m){
        C[x] += d;
        x += lowbit(x);
    }
    return ;
}
void init(){
    memset(C,0,sizeof(C));
    for(int i = m + 1 ; i <= n + m; i++){
        vis[i] = i;
        Add(i,1);
    }
    return ;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        init();
        int ans[MAXD];
        for(int i = 0 ; i < m ; i++){
             int t;
             scanf("%d",&t);
             ans[i] = Sum(vis[t + m])- 1;
             Add(vis[t + m],-1);
             vis[t + m]= m - i;
             Add(vis[t + m],1);
        }
        for(int i = 0 ; i < m ; i++){
            printf("%d",ans[i]);
                if(i < m - 1)
                    printf(" ");
        }
        printf("\n");
    }
    return 0;
}