Codeforces 813E Army Creation [主席树]

最新推荐文章于 2021-01-30 11:43:25 发布

原创最新推荐文章于 2021-01-30 11:43:25 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#主席树 #codeforce

Codeforces 同时被 2 个专栏收录

76 篇文章

订阅专栏

可持久化数据结构

12 篇文章

订阅专栏

本文介绍了一种使用主席树解决区间内元素计数问题的方法。通过构建主席树并结合差分思想，可以有效地处理区间内相同数字最多选择k个的情况。文章提供了详细的AC代码实现。

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

题意：给你长度为n的序列，有q个询问，询问[l,r]区间内相同数字最多选k个的总个数。

题解：建立关于位置的主席树，利用差分的思想，对于新加入的一个数，当前位置+1，并在之前第i-k个位置-1，最后询问root[r]中区间[l,r]的总和。

AC代码：

#include<stdio.h>  
#include<vector>  
#include<algorithm>  
#define N 100005  
using namespace std;  
vector<int>vt[N];  
int tree[N*36],lchild[N*36],rchild[N*36],root[N],tot;  
void insert(int last,int cur,int x,int L,int R,int k)  
{  
    tree[cur]=tree[last]+k;  
    lchild[cur]=lchild[last];  
    rchild[cur]=rchild[last];  
    if(L==R)return ;  
    int mid=L+R>>1;  
    if(x<=mid)insert(lchild[last],lchild[cur]=++tot,x,L,mid,k);  
    else insert(rchild[last],rchild[cur]=++tot,x,mid+1,R,k);  
}  
int query(int l,int r,int L,int R,int root)  
{  
    if(l<=L&&R<=r)return tree[root];  
    int mid=L+R>>1;  
    if(r<=mid)return query(l,r,L,mid,lchild[root]);  
    else if(l>mid)return query(l,r,mid+1,R,rchild[root]);  
    else return query(l,mid,L,mid,lchild[root])+query(mid+1,r,mid+1,R,rchild[root]);  
}  
int main()  
{  
    int n,k,q,x;  
    scanf("%d%d",&n,&k);  
    for(int i=1;i<=n;i++)  
    {  
        scanf("%d",&x);  
        vt[x].push_back(i);  
        insert(root[i-1],root[i]=++tot,i,1,n,1);  
        if(vt[x].size()>k)
		{
			int now=++tot;
			insert(root[i],now,vt[x][vt[x].size()-k-1],1,n,-1);
			root[i]=now;  
		}
    }  
    int last=0;  
    scanf("%d",&q);  
    while(q--)  
    {  
        int l,r;  
        scanf("%d%d",&l,&r);  
        l=(l+last)%n+1;  
        r=(r+last)%n+1;  
        if(l>r)swap(l,r);  
        last=query(l,r,1,n,root[r]);  
        printf("%d\n",last);  
    }  
}