UVa-1513-Movie collection

最新推荐文章于 2019-05-08 14:46:06 发布

原创最新推荐文章于 2019-05-08 14:46:06 发布 · 575 阅读

0 ·

CC 4.0 BY-SA版权

数据结构-树状数组专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用树状数组解决电影光碟查询问题的方法。该问题涉及在每次取出并观看光碟前后更新其位置，并统计未观看光碟的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：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;
}