uva 1513 - Movie collection(树状数组)

来源：互联网发布：软件学徒招聘骗局编辑：程序博客网时间：2024/05/23 11:55

题目连接：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;}

0 0