HDUOJ6602 Longest Subarray(线段树)

HDUOJ6602 Longest Subarray(线段树)

题目大意

给出一段序列，求出其中最长的一段子序列，使得这段子序列中的每个数字出现的次数都要大于等于K次

解题思路

对于每个位置i，维护出这个数字上k次出现的位置，设为lastk[i]（即lastk[i]-i，数字a[i]恰好出现了k次）。枚举右节点，对于每个右节点维护出这个节点之前所有的节点到这个节点形成的区段中有多少的数字出现的次数不到k次。每次右节点向右移动则新位置设为r，就对[last[lastk[i]]+1,lastk[i]]区段-1，对[last[i]+1,i]区段+1，然后再每次查询出最左边的出现次数为0的节点，最后取最大值就是答案

AC代码

#include<bits/stdc++.h>
using namespace std;
const int size=1e5+5;
int n,c,k;
struct node{
    int l,r;
    int tag;
    int minn;
}tree[size<<2];
int arr[size];
int last[size],lastk[size];
vector<int> nums[size];
inline int lson(int x){return x<<1;}
inline int rson(int x){return x<<1|1;}
void build(int id,int l,int r)
{
    tree[id].l=l,tree[id].r=r;
    tree[id].minn=0;
    tree[id].tag=0;
    if(l==r) return ;
    int mid=(l+r)/2;
    build(lson(id),l,mid);
    build(rson(id),mid+1,r);
}
void pushdown(int id)
{
    if(tree[id].l!=tree[id].r)
    {
        tree[lson(id)].minn=tree[lson(id)].minn+tree[id].tag;
        tree[lson(id)].tag=tree[lson(id)].tag+tree[id].tag;
        tree[rson(id)].minn=tree[rson(id)].minn+tree[id].tag;
        tree[rson(id)].tag=tree[rson(id)].tag+tree[id].tag;
    }
    tree[id].tag=0;
}
void update(int id,int ul,int ur,int v)
{
    if(tree[id].tag) pushdown(id);
    if(tree[id].l==ul&&tree[id].r==ur)
    {
        tree[id].minn+=v;
        tree[id].tag+=v;
        return ;
    }
    int mid=(tree[id].l+tree[id].r)>>1;
    if(mid>=ur) update(lson(id),ul,ur,v);
    else if(mid<ul) update(rson(id),ul,ur,v);
    else update(lson(id),ul,mid,v),update(rson(id),mid+1,ur,v);
    tree[id].minn=min(tree[lson(id)].minn,tree[rson(id)].minn);
}
inline int query(int id,int ql,int qr)
{
    if(tree[id].l>qr) return qr+1;
    if(tree[id].tag!=0) pushdown(id);
    if(tree[id].l==tree[id].r) return tree[id].l;
    int mid=(tree[id].l+tree[id].r)>>1;
    int ans=qr+1;
    if(tree[lson(id)].minn==0) ans=query(lson(id),ql,qr);
    else if(tree[rson(id)].minn==0) ans=query(rson(id),ql,qr);
    tree[id].minn=min(tree[lson(id)].minn,tree[rson(id)].minn);
    return ans;
}
int main()
{
    while(~scanf("%d%d%d",&n,&c,&k))
    {
        for(int i=1;i<=n;i++) scanf("%d",&arr[i]);
        if(k==1)
        {
            printf("%d\n",n);
            continue;
        } 
        for(int i=1;i<=n;i++)
        {
            int cnts=nums[arr[i]].size();
            if(cnts>=1) last[i]=nums[arr[i]][cnts-1];else last[i]=0;
            if(cnts>=k-1) lastk[i]=nums[arr[i]][cnts-k+1];else lastk[i]=0;
            nums[arr[i]].push_back(i);
        }
        for(int i=1;i<=c;i++) nums[i].clear();
        build(1,1,n);
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            update(1,last[i]+1,i,1);
            if(lastk[i]!=0)
            update(1,lastk[last[i]]+1,lastk[i],-1);
            int l=query(1,1,i);
            ans=max(ans,i-l+1);
        }
        printf("%d\n",ans);
    }
}