【CF813E】Army Creation（主席树）

最新推荐文章于 2020-07-07 12:56:56 发布

小蒟蒻yyb

最新推荐文章于 2020-07-07 12:56:56 发布

阅读量426

点赞数 1

CC 4.0 BY-SA版权

分类专栏： CF 主席树

本文链接：https://blog.youkuaiyun.com/qq_30974369/article/details/79795004

主席树同时被 2 个专栏收录

19 篇文章

订阅专栏

15 篇文章

订阅专栏

本文介绍了一种使用主席树解决区间查询问题的方法，并通过一个具体题目实例详细讲解了实现过程。文章首先定义了问题背景，即在一个数组中进行特定条件下的区间查询，接着介绍了如何构建主席树以及如何进行区间修改和查询操作。最后通过代码示例展示了完整的解决方案。

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

题面

CF
洛谷
翻译 by ppl 见洛谷

题解

考虑最多只会有 $K$ 个相同的数
那么，也就是说，如果一个数会被选
那么，和它相等的数中，在它前面的第 $K$ 的位置应该小于 $l$
所以开个 $vector$ 直接搞
然后按照前面的第 $K$ 个数的位置插入主席树
查询 $0..l-1$ 的值即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 111111
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Node
{
    int ls,rs;
    int v;
}t[MAX<<5];
int tot,rt[MAX];
void Modify(int &x,int ff,int l,int r,int p,int w)
{
    t[x=++tot]=t[ff];t[x].v+=w;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(p<=mid)Modify(t[x].ls,t[ff].ls,l,mid,p,w);
    else Modify(t[x].rs,t[ff].rs,mid+1,r,p,w);
}
int Query(int r1,int r2,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)return t[r1].v-t[r2].v;
    int mid=(l+r)>>1,ret=0;
    if(L<=mid)ret+=Query(t[r1].ls,t[r2].ls,l,mid,L,R);
    if(R>mid)ret+=Query(t[r1].rs,t[r2].rs,mid+1,r,L,R);
    return ret;
}
vector<int> S[MAX];
int lst[MAX],n,K,a[MAX];
int main()
{
    n=read();K=read();
    for(int i=1;i<=n;++i)a[i]=read();
    for(int i=1;i<=n;++i)
    {
        if(S[a[i]].size()>=K)lst[i]=S[a[i]][S[a[i]].size()-K];
        S[a[i]].push_back(i);
    }
    for(int i=1;i<=n;++i)Modify(rt[i],rt[i-1],0,n,lst[i],1);
    int Q=read(),ans=0;
    while(Q--)
    {
        int l=(read()+ans)%n+1,r=(read()+ans)%n+1;
        if(l>r)swap(l,r);
        printf("%d\n",ans=Query(rt[r],rt[l-1],0,n,0,l-1));
    }
    return 0;
}