2019HDU多校第二場 HDU-6602 Longest Subarray (線段樹 + 思維)

題意： 長度爲n的序列，求最大的子序列長度，要求子序列中所出現的數字個數>=k。

思路： 枚舉右邊界r，線段樹維護左邊界l的範圍。
對於每一個數a[r]來說，我們可以清楚的知道 l 可以在什麼地方

放入一個 a[r] 對於 i 位置 c - 1數據不需要出現
對於它之前出現的 我們是要選擇r這個位置的數據的 所以我們要把 它前一個數據位置 到 r - 1 先 -1
選擇r位置 就把之前位置在的地方a[r] 數據出現減去

離a[r]最近的 同一個數據數子 位置爲 P1，
離a[r]第k遠的 同一個數據數字 位置 P2，
它這個數據之後的下一位 P3
當 a[r] 加入 p2 ~ p3 位置 就對a[r] 數據 合法了 區間 + 1；
右邊 就是 p1 ~ r 這個區間 與之前 1 ~ p2 ~ p3 區間 對應
最後查詢區間個數>=c的最左邊的邊界l即可。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;

int tree[maxn << 2], lazy[maxn << 2];
vector<int> vec[maxn];
int n, C, k;
int a[maxn], pos[maxn];

void push_up(int rt) {
    tree[rt] = max(tree[rt << 1], tree[rt << 1 | 1]);
}

void push_down(int rt) {
    if(lazy[rt] == 0) return ;
    tree[rt << 1] += lazy[rt];
    lazy[rt << 1] += lazy[rt];
    tree[rt << 1 | 1] += lazy[rt];
    lazy[rt << 1 | 1] += lazy[rt];
    lazy[rt] = 0;
}

void build(int l, int r, int rt) {
    tree[rt] = lazy[rt] = 0;
    if(l == r) return ;
    int mid = l + r >> 1;
    build(l, mid, rt << 1);
    build(mid + 1, r, rt << 1 | 1);
}

void updata(int L, int R, int l, int r, int rt, int val) {
    if(L > R) return ;
    if(L <= l && R >= r) {
        tree[rt] += val;
        lazy[rt] += val;
        return ;
    }
    push_down(rt);
    int mid = l + r >> 1;
    if(L <= mid) updata(L, R, l, mid, rt << 1, val);
    if(R > mid) updata(L, R, mid + 1, r, rt << 1 | 1, val);
    push_up(rt);
}

int query(int l, int r, int rt) {
    if(tree[rt] < C) return 0;
    if(l == r) return l;
    push_down(rt);
    int mid = l + r >> 1;
    if(tree[rt << 1] >= C) return query(l, mid, rt << 1);
    if(tree[rt << 1 | 1] >= C) return query(mid + 1, r, rt << 1 | 1) ;
}

int main() {
    while(~scanf("%d%d%d", &n, &C, &k)) {
        for(int i = 1; i <= C; i++)
            vec[i].clear(), vec[i].emplace_back(0);
        int ans = 0;
        build(1, n, 1);
        for(int i = 1, x; i <= n; i++) {
            scanf("%d", &x);
            updata(i, i, 1, n, 1, C - 1);
            updata(vec[x].back() + 1, i - 1, 1, n, 1, -1);
            vec[x].emplace_back(i);
            int t = vec[x].size() - k - 1;
            if(t >= 0) updata(vec[x][t] + 1, vec[x][t + 1], 1, n, 1, 1);
            int j = query(1, n, 1);
            if(j) ans = max(ans, i - j + 1);
        }
        cout << ans << endl;
    }
    return 0;
}