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