牛客练习赛73.D.离别(线段树+离线)

牛客练习赛73.D.离别(线段树+离线)

题意：

给定一个长度为n的数列，每次询问一个区间(L, R)，求满足区间内最大出现次数等于k次的子区间个数

思路：

1.预处理：对于每个节点，以其为右区间端点，设$[li,ri]$为其左区间端点的取值范围，此区间满足出现对多的数为 $k$。可以用队列存储每个数字出现的位置，当队列长度等于$k$时，更新$r_i$，当队列长度大于$k$时，更新$l_i$，再更新$r_i$。可以$O(n)$实现。

2.离线询问：在以$q$次询问的区间的右边界从小到大排序，枚举当前区间时，若q_ri其右区间端点为，则先把上述预处理的，在q_ri左边的点的$[l,r]$，区间+1，询问则转化成查询区间值和，可以用线段树维护

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+5;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
int n,m,k,a[maxn],l[maxn],r[maxn];
ll ans[maxn],lazy[maxn<<2],sum[maxn<<2];
queue<int>que[maxn];
struct node{
	int l,r,id;
}p[maxn];
bool cmp(node x,node y){
	return x.r<y.r;
}
void pushdown(int rt,int l,int r){
    if(lazy[rt]){
	    int mid=(l+r)>>1;
	    lazy[rt<<1]+=lazy[rt];
	    sum[rt<<1]+=1ll*(mid-l+1)*lazy[rt];
	    lazy[rt<<1|1]+=lazy[rt];
	    sum[rt<<1|1]+=1ll*(r-mid)*lazy[rt];
        lazy[rt]=0;
	}
}
void update(int rt,int l,int r,int x,int y,int v){
    if(x<=l&&r<=y){
       lazy[rt]+=v;
       sum[rt]+=1ll*v*(r-l+1);
       return;
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    if(x<=mid)
      update(lson,x,y,v);
    if(mid<y)
      update(rson,x,y,v);
     sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
ll query(int rt,int l,int r,int x,int y){
    if(x<=l&&r<=y)return sum[rt];
    int mid=(l+r)>>1;
	ll ans=0;
    pushdown(rt,l,r);
    if(x<=mid)
      ans+=query(lson,x,y);
    if(mid<y)
       ans+=query(rson,x,y);
    return ans;     
}
int main(){
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=m;i++){
		scanf("%d%d",&p[i].l,&p[i].r);
		p[i].id=i;
	}
	l[0]=1;
	for(int i=1;i<=n;i++){
		l[i]=l[i-1],r[i]=r[i-1];
		que[a[i]].push(i);
		if(que[a[i]].size()>k){
			l[i]=max(l[i-1],que[a[i]].front()+1);
			que[a[i]].pop();
		}
		if(que[a[i]].size()==k){
			r[i]=max(r[i-1],que[a[i]].front());
		}
	}
	sort(p+1,p+1+m,cmp);
	int j=1;
	for(int i=1;i<=m;i++){
	    while(j<=p[i].r){
	    	if(r[j])update(1,1,n,l[j],r[j],1);
	    	j++;
		}
		ans[p[i].id]=query(1,1,n,p[i].l,p[i].r);
	}
	for(int i=1;i<=m;i++)printf("%lld\n",ans[i]);
}