2019 멀티 - 대학 교육 경연 대회 2 - 가장 긴 부분 배열

세그먼트 트리

모든 법적 커버의 범위를 나타내며, 마지막으로 최소 위치에 대한 최대 값을 찾기 위해 플러스 트리 라인의 범위와 오른쪽 지점을 열거하면 왼쪽 C 점이다

편의상, 각 번호의 첫번째 발생의 위치가 마지막으로 N + 1과 0으로한다.

필요 간격 -1 덮어, 우리는 I에 열거 나이 위치 선택해야 번호 이후의 최초 발생에 위치 0으로 시작하기 때문에, 회수 그는 단지 원래 같거나 K 커야 등장 간격은 차감 대표되는 표시되지 않습니다.

이후 발생 회수는, 제 k 구간 증가 해당 출현 다시 트리 라인을 업데이트 할 필요성을 증가시킨다.

대답은 모든 통계를 열거 할 때마다, 중은 크다.

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define full(a, b) memset(a, b, sizeof a)
#define __fastIn ios::sync_with_stdio(false), cin.tie(0)
#define pb push_back
using namespace std;
typedef long long LL;
inline int lowbit(int x){ return x & (-x); }
inline int read(){
    int ret = 0, w = 0; char ch = 0;
    while(!isdigit(ch)){
        w |= ch == '-', ch = getchar();
    }
    while(isdigit(ch)){
        ret = (ret << 3) + (ret << 1) + (ch ^ 48);
        ch = getchar();
    }
    return w ? -ret : ret;
}
inline int lcm(int a, int b){ return a / __gcd(a, b) * b; }
template <typename A, typename B, typename C>
inline A fpow(A x, B p, C lyd){
    A ans = 1;
    for(; p; p >>= 1, x = 1LL * x * x % lyd)if(p & 1)ans = 1LL * x * ans % lyd;
    return ans;
}
const int N = 200005;
int n, c , k, a[N], tree[N<<2], lazy[N<<2], freq[N];
vector<int> vi[N];

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

void f(int rt, int val){
    tree[rt] += val, lazy[rt] += val;
}

void push_down(int rt, int l, int r){
    int mid = (l + r) >> 1;
    f(rt << 1, lazy[rt]), f(rt << 1 | 1, lazy[rt]);
    lazy[rt] = 0;
}

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

void add(int rt, int l, int r, int al, int ar, int val){
    if(al > ar) return;
    if(l == al && r == ar){
        f(rt, val);
        return;
    }
    push_down(rt, l, r);
    int mid = (l + r) >> 1;
    if(ar <= mid) add(rt << 1, l, mid, al, ar, val);
    else if(al > mid) add(rt << 1 | 1, mid + 1, r, al, ar, val);
    else add(rt << 1, l, mid, al, mid, val), add(rt << 1 | 1, mid + 1, r, mid + 1, ar, val);
    push_up(rt);
}

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

int main(){

    __fastIn;

    while(cin >> n >> c >> k){
        full(freq, 0), full(lazy, 0), full(tree, 0);
        for(int i = 1; i <= c; i ++){
            vi[i].clear();
            vi[i].pb(0);
        }
        for(int i = 1; i <= n; i ++){
            cin >> a[i];
            vi[a[i]].pb(i);
        }
        for(int i = 1; i <= c; i ++){
            vi[i].pb(n + 1);
        }
        buildTree(1, 1, n);
        for(int i = 1; i <= c; i ++){
            add(1, 1, n, 1, vi[i][1] - 1, 1);
        }
        int ans = 0;
        for(int i = 1; i <= n; i ++){
            add(1, 1, n, vi[a[i]][freq[a[i]]] + 1, vi[a[i]][freq[a[i]] + 1] - 1, -1);
            freq[a[i]] ++;
            add(1, 1, n, vi[a[i]][freq[a[i]]] + 1, vi[a[i]][freq[a[i]] + 1] - 1, 1);
            if(freq[a[i]] >= k) add(1, 1, n, vi[a[i]][freq[a[i]] - k] + 1, vi[a[i]][freq[a[i]] - k + 1], 1);
            int tmp = query(1, 1, n);
            if(tmp !=  -1) ans = max(ans, i - tmp + 1);
        }
        printf("%d\n", ans);
    }
    return 0;
}

추천

출처www.cnblogs.com/onionQAQ/p/11274257.html