HDU-6602 Longest Subarray（线段树）

最新推荐文章于 2020-07-03 18:29:35 发布

A_Thinking_Reed_

最新推荐文章于 2020-07-03 18:29:35 发布

阅读量133

点赞数

分类专栏：数据结构 | 线段树

本文链接：https://blog.csdn.net/A_Thinking_Reed_/article/details/99474202

版权

数据结构 | 线段树专栏收录该内容

15 篇文章 0 订阅

订阅专栏

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

题意：数列中的数最大为c，求数列的最长子串，满足子串中的数字出现至少 k 次或0次。

思路：枚举子串右端点，对于当前数来说，满足出现至少k次或0次的左端点有范围[1, lastk[i]]（k次及以上）, [last[i]+1, i-1]（0次），每次将满足条件的区间+1，最后找最左端的 c。每次移动右端点时维护合法区间。

#include<bits/stdc++.h>
#define ls rt<<1
#define rs rt<<1|1
using namespace std;
const int maxn = 1e5+5;

int Max[maxn<<2], tag[maxn<<2];
void pushdown(int rt){
    if(tag[rt] != 0){
        tag[ls] += tag[rt]; tag[rs] += tag[rt];
        Max[ls] += tag[rt]; Max[rs] += tag[rt];
        tag[rt] = 0;
    }
}
void pushup(int rt){Max[rt] = max(Max[ls], Max[rs]);}
void build(int rt, int l, int r){
    Max[rt] = tag[rt] = 0;
    if(l == r) return ;
    int mid = l+r >> 1;
    build(ls, l, mid); build(rs, mid+1, r); pushup(rt);
}
void update(int rt, int l, int r, int L, int R, int val){
    if(L > R || l > r) return ;
    if(L <= l && R >= r) {
        Max[rt] += val, tag[rt] += val;
        return ;
    }
    pushdown(rt);
    int mid = l+r >> 1;
    if(L <= mid) update(ls, l, mid, L, R, val);
    if(R > mid) update(rs, mid+1, r, L, R, val);
    pushup(rt);
}
int query(int rt, int l, int r, int val){
    if(l == r) return l;
    pushdown(rt);
    int mid = l+r >> 1, ans = -1;
    if(Max[ls] == val) ans = query(ls, l, mid, val);
    else if(Max[rs] == val) ans = query(rs, mid+1, r, val);
    return ans ;
}
int n, c, k, a[maxn], now[maxn];
vector<int>vec[maxn];
int main(){
    while(~scanf("%d%d%d", &n, &c, &k)){
        for(int i=1; i<=c; i++) vec[i].clear(), vec[i].push_back(0);
        for(int i=1; i<=n; i++) {
            scanf("%d", &a[i]);
            vec[a[i]].push_back(i);
        }
        build(1, 1, n);
        for(int i=1; i<=c; i++) {
            vec[i].push_back(n+1); now[i] = 0;
            update(1, 1, n, vec[i][0]+1, vec[i][1]-1, 1);
        }
        int ans = 0;
        for(int i=1; i<=n; i++){
            int t = a[i];
            update(1, 1, n, vec[t][now[t]]+1, vec[t][now[t]+1]-1, -1);
            if(now[t] >= k) update(1, 1, n, 1, vec[t][now[t]-k+1], -1);
            now[t]++;
            update(1, 1, n, vec[t][now[t]]+1, vec[t][now[t]+1]-1, 1);
            if(now[t] >= k) update(1, 1, n, 1, vec[t][now[t]-k+1], 1);
            int tmp = query(1, 1, n, c);
            if(tmp != -1) ans = max(ans, i-tmp+1);
        }
        cout << ans << endl;
    }
}