Movie collection UVA

来源：互联网发布：一本好书知乎编辑：程序博客网时间：2024/05/22 03:06

学习时参考了这篇文章，看完基本就能弄懂树状数组是怎么回事了：http://blog.csdn.net/int64ago/article/details/7429868

题目链接：https://vjudge.net/contest/180021#problem/D

题目描述：一开始共有编号1~n共n张影片从上之下排列，每次抽出一张影片x，求抽出时影片x上面有几张影片，每次将抽出的影片放在最上面。

思路：搜了题解说是树状数组，然后就去网上学，说真的，真的不怪别人瞧不起自己是我自己能力不高，之前一年多时间没把心思主要放在ACM上，连树状数组模板题都写不出来，现在除了让自己变强，别的都是浮云，什么都不想考虑但求问心无愧。一不小心就跑题了，还是赶紧回来做正事吧。每次抽出一张影片x放到最上面，它原来位置下面的影片不会受影响，上面的影片在它们上面的影片数会加1（因为x放到了它们最上面），所以考虑将影片逆序排列，即最下面的位置是1，最上面是n，这样就可以用树状数组来维护。

代码如下：

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#include<queue>#include<cstdlib>#include<sstream>#include<deque>#include<stack>#include<set>#include<map>using namespace std;typedef long long ll;typedef unsigned long long ull;const double eps = 1e-6;const int  maxn = 200000 + 10;const int  maxt = 300 + 10;const int mod = 10;const int dx[] = {1, -1, 0, 0};const int dy[] = {0, 0, -1, 1};const int Dis[] = {-1, 1, -5, 5};const int inf = 0x3f3f3f3f;const int MOD = 1000;int n, m, k;int tree[maxn], pos[maxn];inline int lowbit(int x){//求x只留下最低位的1后所得数字，如：lowbit(100110) = 10, lowbit(1000) = 1000    return x & -x;}void update(int x, int num){//x位置的值增加num，维护新的树状数组值    while(x < maxn){//修改的是x及其之后整个区间        tree[x] += num;        x += lowbit(x);    }}int getsum(int x){//求区间[1,x]的区间和    int sum = 0;    while(x >= 1){        sum += tree[x];        x -= lowbit(x);    }    return sum;}int main(){    int t;    scanf("%d", &t);    while(t--){        memset(tree, 0, sizeof tree);        scanf("%d%d", &n, &m);        for(int i = 1; i <= n; ++i){            pos[i] = n - i + 1;//1~n倒序排列            update(i, 1);        }        bool ok = false;        int x;        int cnt = n;        for(int i = 1; i <= m; ++i){            scanf("%d", &x);            if(ok) printf(" ");            else ok = true;            printf("%d", n - getsum(pos[x]));//x之上有多少影片            update(pos[x], -1);//要将x抽出，故x上面的所有位置的值减一，因为他们下面少了一张影片            pos[x] = ++cnt;//x跑到最上面，倒序排列即tree数组最后面            update(pos[x], 1);//更新最后面的位置值        }        printf("\n");    }    return 0;}

阅读全文

0 0