题意:Mr. K先生有n本电影光碟放在一个栈里,编号从小到大从上到下的排着,每当他想看的时候他就拿出来(看了m次),看完后把它放在栈顶,问你他在每次没看这部电影之前的上面有多少本电影。
思路:就是普通的树状数组只不过就是要前面多开出m个空间来,还要开个数组来存对应编号的电影在哪个位置上。
AC代码:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=100100;
int l[maxn<<1],vis[maxn<<1]; //l是树状数组 vis是保存第i本电影的位置
int t,n,m;
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int d)
{
while(x<=n+m)
{
l[x]+=d;
x+=lowbit(x);
}
}
void init()
{
for(int i=m+1;i<=m+n;i++)
{
add(i,1);
vis[i]=i;
}
}
int sum(int x)
{
int cnt=0;
while(x)
{
cnt+=l[x];
x-=lowbit(x);
}
return cnt;
}
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
memset(vis,0,sizeof(vis));
memset(l,0,sizeof(l));
init();
int ans[maxn]; //保存结果的
for(int i=0;i<m;i++)
{
int x;
scanf("%d",&x);
ans[i]=sum(vis[m+x])-1; //这必须减一 因为算了它本身
add(vis[m+x],-1);
vis[m+x]=m-i; //把看完的电影依次往前放
add(vis[m+x],1);
}
for(int i=0;i<m-1;i++)
printf("%d ",ans[i]);
printf("%d\n",ans[m-1]);
}
return 0;
}