Movie collection UVA - 1513(树状数组模板题)

学习时参考了这篇文章,看完基本就能弄懂树状数组是怎么回事了: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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值