2019 Multi-University Training Contest 2 1012 Longest Subarray —— 线段树求每个位置出现次数>=k或0的最长区间

This way

题意：

给你n个数，问你其中最长连续子区间中每个数出现次数为0或者>=k次的区间长度是多少

题解：

枚举右端点，到当前位置为止，c这个数有两种可能的合法区间：
1.距离当前端点左边第一个c+1的位置到当前端点
2.1到距离当前端点前k个c的位置
那么我们每到一个新的位置，它的值是x，我们首先将这个位置的值赋为c-1，因为第一种情况不可能出现，然后将上一个x出现的位置+1到这个位置-1的区间中的值都-1，因为这些位置也缺少了这个数。之后将前k个位置+1到前k-1个位置的区间+1，因为这些位置数x是合法的。
那么我们查询的时候只需要尽量往左区间查即可。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
vector<int>vec[N];
int a[N],mx[N*4],flag[N*4];
int n,c,k;
void push_down(int root)
{
    if(!flag[root])
        return ;
    flag[root<<1]+=flag[root];
    flag[root<<1|1]+=flag[root];
    mx[root<<1]+=flag[root];
    mx[root<<1|1]+=flag[root];
    flag[root]=0;
}
void update(int l,int r,int root,int ql,int qr,int v)
{
    if(l>=ql&&r<=qr)
    {
        mx[root]+=v;
        flag[root]+=v;
        return ;
    }
    push_down(root);
    int mid=l+r>>1;
    if(mid>=ql)
        update(l,mid,root<<1,ql,qr,v);
    if(mid<qr)
        update(mid+1,r,root<<1|1,ql,qr,v);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
int query(int l,int r,int root)
{
    if(l==r)
        return l;
    push_down(root);
    int mid=l+r>>1;
    if(mx[root<<1]==c)
        return query(l,mid,root<<1);
    if(mx[root<<1|1]==c)
        return query(mid+1,r,root<<1|1);
    return N;
}
int main()
{

    while(~scanf("%d%d%d",&n,&c,&k))
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        if(k==1)
        {
            printf("%d\n",n);
            continue;
        }
        for(int i=1;i<=c;i++)
            vec[i].clear(),vec[i].push_back(0);
        for(int i=1;i<=n*4;i++)
            mx[i]=flag[i]=0;
        int cnt[N]={0},ans=0;
        for(int i=1;i<=n;i++)
        {
            update(1,n,1,i,i,c-1);
            cnt[a[i]]++;
            vec[a[i]].push_back(i);
            if(vec[a[i]][cnt[a[i]]-1]+1<=vec[a[i]][cnt[a[i]]]-1)
                update(1,n,1,vec[a[i]][cnt[a[i]]-1]+1,vec[a[i]][cnt[a[i]]]-1,-1);
            if(cnt[a[i]]>=k)
                update(1,n,1,vec[a[i]][cnt[a[i]]-k]+1,vec[a[i]][cnt[a[i]]-k+1],1);
            int l=query(1,n,1);
            if(l!=N)
                ans=max(ans,i-l+1);
        }
        printf("%d\n",ans);
    }
    return 0;
}