19 hdu多校 Longest Subarray //线段树

19 hdu多校 Longest Subarray //线段树

http://acm.hdu.edu.cn/showproblem.php?pid=6602

题意：给长度为n的数组，取最长的长度为k的subarray，使得每个元素要么不出现，要么出现次数大于k。

题解思路：

：

我相反，线段树维护的是不能取的区间+1。然后询问的时候，查找最左边，最小值为0的叶子节点的下标。

vector记录加入不可取区间的元素，真的可以优化很多啊 -。- 自己写的好杂啊

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define FI first
#define SE second
#define MP make_pair
#define PII pair<int,int>
const int MXLOG = 30;
const LL mod = 998244353;
const int MX = 1e5+5;
int T[MX*4],lazy[MX*4];//维护区间最小
int lineL[MX],lineR[MX];//维护每一个元素不可取的区间
int nx[MX],fito[MX];//第i个元素的后继，和l第一跳到的位置
int a[MX],last[MX],cnt[MX];
void pushUp(int root){
    T[root]=min(T[root<<1],T[root<<1|1]);
}
void pushDown(int root){
    if(lazy[root]){
        lazy[root<<1]+=lazy[root];
        lazy[root<<1|1]+=lazy[root];
        T[root<<1]+=lazy[root];
        T[root<<1|1]+=lazy[root];
        lazy[root]=0;
    }
}
void update(int L,int R,int add,int l,int r,int root){
    if(L<=l&&r<=R) {
        lazy[root]+=add;
        T[root]+=add;
        return ;
    }
    pushDown(root);
    int m=(l+r)/2;
    if(m>=L) update(L,R,add,l,m,root<<1);
    if(m<R)  update(L,R,add,m+1,r,root<<1|1);
    pushUp(root);
}
int query(int l,int r,int root){
    if(l==r)return T[root]!=0?1e6:l;
    pushDown(root);
    int mid=(l+r)/2;
    if(T[root<<1]==0) query(l,mid,root<<1);
    else query(mid+1,r,root<<1|1);
}
int main(){
    int n,c,k;while(cin>>n>>c>>k){
        for(int i=1;i<=n;i++)scanf("%d",&a[i]),nx[i]=0;
        for(int i=0;i<=4*n;i++)lazy[i]=0,T[i]=0;
        if(k==0||k==1){cout<<n<<'\n';continue;}
        for(int i=1;i<=c;i++) last[i]=cnt[i]=0,lineL[i]=1,lineR[i]=fito[i]=0;
        for(int i=n;i>=0;i--){
            nx[i]=last[a[i]];
            last[a[i]]=i;
        }
        for(int i=1;i<=c;i++)fito[i]=last[i]; //第一次都在0,确定后继
        int ans=0;
        for(int i=1;i<=n;i++){
            if(lineL[a[i]]<=lineR[a[i]])
                update(lineL[a[i]],lineR[a[i]],-1,1,n,1);
            cnt[a[i]]++;lineR[a[i]]=i;
            if(cnt[a[i]]==k){
                cnt[a[i]]--;
                if(fito[a[i]])lineL[a[i]]=fito[a[i]]+1,fito[a[i]]=0;
                else lineL[a[i]]=nx[lineL[a[i]]-1]+1;
            }
            if(lineL[a[i]]<=lineR[a[i]])
                update(lineL[a[i]],lineR[a[i]],1,1,n,1);
            int zz=query(1,n,1);
            ans=max(ans,1+i-zz);
        }
        printf("%d\n",ans);
    }
    return 0;
}