codeforces 813E 主席树

题意:

一个数列多组询问,每次询问[l,r]中最多能选多少个数字,其中每个数字的出现次数不超过k次

题解:

我们保存对于每个位置上,出现超过k次的位置,那么对于每次询问,我们就变成了查询区间[l,r]大于r的数字个数

可以离线,但是本题强制在线,因此使用主席树

#include <bits/stdc++.h>
#define endl '\n'
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define rep(ii,a,b) for(int ii=a;ii<=b;++ii)
#define per(ii,a,b) for(int ii=b;ii>=a;--ii)
using namespace std;
const int maxn=1e5+10,maxm=2e6+10;
//head
int casn,n,m,k;
int rt[maxn],num[maxn];
class ptree{public:
#define nd node[now]
#define ndp node[pre]
#define mid (s+t)/2
	struct segnode{int l,r,sum;}node[maxn*30];
	int cnt;
	void maketree(int s,int t,int &now=rt[0]){
    now=++cnt;nd={s,t,0};
    if(s==t) return ;
    maketree(s,mid,nd.l); maketree(mid+1,t,nd.r);
	}
	void update(int pos,int val,int s,int t,int &now,int pre){
    now=++cnt;nd=ndp;nd.sum+=val;
    if(s==t) return ;
    if(pos<=mid) update(pos,val,s,mid,nd.l,ndp.l);
    else update(pos,val,mid+1,t,nd.r,ndp.r);
	}
	int query(int l,int r,int s,int t,int now,int pre){
    if(l<=s&&r>=t) return nd.sum-ndp.sum;
    if(l>mid) return query(l,r,mid+1,t,nd.r,ndp.r);
    else if(r<=mid) return query(l,r,s,mid,nd.r,ndp.r);
    else return query(l,r,s,mid,nd.l,ndp.l)+query(l,r,mid+1,t,nd.r,ndp.r);
	}
}tree;
vector<int>pos[maxn];
int a[maxn];
int main() {
IO;
  cin>>n>>m;
  rep(i,1,n) cin>>num[i];
  per(i,1,n){
    int sz=pos[num[i]].size();
    if(sz<m) a[i]=n+1;
    else a[i]=pos[num[i]][sz-m];
    pos[num[i]].push_back(i);
  }
  tree.maketree(1,n+1);
  rep(i,1,n) tree.update(a[i],1,1,n+1,rt[i],rt[i-1]);
	cin>>k;
	int ans=0;
	while(k--){int l,r;
    cin>>l>>r;
    l=(l+ans)%n+1;
    r=(r+ans)%n+1;
    if(l>r) swap(l,r);
    ans=tree.query(r+1,n+1,1,n+1,rt[r],rt[l-1]);
    cout<<ans<<endl;
	}
	return 0;
}

 

转载于:https://www.cnblogs.com/nervendnig/p/10759268.html