CodeForces 813E :Army Creation 主席树

传送门

题目描述

给你一个长度为$n$序列$a_i$，$q$次询问，每次给你一个$l,r$，问你在$[l,r]$内最多可以选多少个数,满足同一个数的出现次数不超过$k$
强制在线

分析

我们去存一下每一个数字出现的位置，就可以倒推前面$k$个数出现的位置，也就是说只要前面$k$个数字出现的位置大于$l$都可以选，然后我们可以用主席树储存一下每一段可选的个数，然后区间操作即可

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod= 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a){char c=getchar();T x=0,f=1;while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}a=f*x;}
int gcd(int a,int b){return (b>0)?gcd(b,a%b):a;}
struct Node
{
	int l,r;
	int cnt;
}tr[N * 60];
int n,m,k;
int idx;
int root[N];
vector<int> nums[N];

int insert(int p,int l,int r,int num){
	int q = ++idx;
	tr[q] = tr[p];
	tr[q].cnt++;
	if(l == r) return q;
	int mid = l + r >> 1;
	if(num <= mid) tr[q].l = insert(tr[p].l,l,mid,num);
	else tr[q].r = insert(tr[p].r,mid + 1,r,num);
	return q;
}

int query(int L,int R,int l,int r,int num){
	if(R <= num) return tr[r].cnt - tr[l].cnt;
	int mid = L + R >> 1;
	int ans = 0;
	ans = query(L,mid,tr[l].l,tr[r].l,num);
	if(num > mid) ans += query(mid + 1,R,tr[l].r,tr[r].r,num);
	return ans;
}

int main(){
    read(n),read(k);
    for(int i = 1;i <= n;i++){
    	int x;
    	read(x);
    	nums[x].pb(i);
    	int sz = SZ(nums[x]);
    	if(sz > k) sz = nums[x][sz - k - 1];
    	else sz = 0;
    	root[i] = insert(root[i - 1],0,n,sz);
    }
    int last = 0;
    read(m);
    while(m--){
    	int l,r;
    	read(l),read(r);
    	l = (l + last) % n + 1, r = (r + last) % n + 1;
    	if(l > r) swap(l,r);
    	last = query(0,n,root[l - 1],root[r],l - 1);
    	di(last);
    } 
    return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/