UVa1513/LA5902 Movie collection

转载

UVA1513 - Movie collection

题目链接

题目大意：给你一叠DV，编号1到n，1在最上面，n在最下面。然后现在给你m个操作，每次都指定一张CD，问要拿走这个CD需要挪走上面多少张CD，并且这个要拿走的CD放在这个叠CD的顶端。

解题思路：将这些CD的上面有多少张CD用数状数组来存储，但是每次要移动的话就有点难办，所以每次移动就将这张CD的pos更新一下，将之前的pos作废，以后查询也是用之后的pos。

代码：

#include <cstdio>
#include <cstring>
 
#define lowbit(x) ((x)&(-x))
 
const int maxn = 1e5 + 5;
 
int c[2 * maxn], pos[maxn];
int n, m;
 
void add (int p, int v) {
 
    while (p < 2 * maxn) {
 
        c[p] += v;
        p += lowbit(p);
    }
}
 
int sum (int p) {
 
    int ret = 0;
    while (p > 0) {
 
        ret += c[p];
        p -= lowbit(p);
    }
    return ret;
}
 
int main () {
 
    int T;
    scanf ("%d", &T);
 
    while (T--) {
 
        memset (c, 0, sizeof (c));
        scanf ("%d%d", &n, &m);
 
        for (int i = 1; i <= n; i++) {
 
            pos[i] = n - i + 1;
            add(pos[i], 1);
        }
 
        int num;
        int N = n;
 
        for (int i = 1; i <= m; i++) {
 
            scanf ("%d", &num);    
            printf ("%d%c", n - sum(pos[num]), i == m ? '\n':' ');    
            add(pos[num], -1);
            pos[num] = ++N;
            add(pos[num], 1);                
        }
    }
    return 0;
}