2019杭电多校 HDU - 6602 Longest Subarray 线段树 标记更新

题目链接：https://vjudge.net/problem/HDU-6602

题意：求一个最长的区间，满足每个数的数量要么大于k，要么为0

题解：

官方题解：如果右端点固定，对于每种元素，可行的左端点下标是两段连续的区间。对于每种元素，将它的可行左端点区间在线段树中加一。 当右端点右移的时候，维护C 种元素的可行左端点。 查询时只需要询问线段树中最小的、值为C 的下标即可。

我是用vector记录下了每个数的位置，方便在操作的时候查询当前这个数是第几个。注意下细节，维护下每个数的可行区间即可

分享一下几个错的思路，一个是二分，细想一下，这个题显然不符合二分的性质，另一个想到了分治，然后一直T，最后想到一些情况就是，对于一个区间每次去掉的数很少的情况，复杂度就能达到n^2 / k，这也是不能过的，但是可能是因为数据水吧，写了一个错的分治，居然过了。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
struct node {
    int l, r; 
    int val, laz;
}tree[N << 2];
int n, c, k;
int a[N];
vector<int> v[N];
void pushup(int cur) {
    tree[cur].val = max(tree[cur << 1].val, tree[cur << 1 | 1].val);
}
void build(int l, int r, int cur) {
    tree[cur].l = l;
    tree[cur].r = r;
    tree[cur].val = 0;
    tree[cur].laz = 0;
    if(l == r) {
        if(l == 1) tree[cur].val = c - 1;
        return;
    }
    int mid = (r + l) >> 1;
    build(l, mid, cur << 1);
    build(mid + 1, r, cur << 1 | 1);
    pushup(cur);
}
void pushdown(int cur) {
    if(tree[cur].laz) {
        tree[cur << 1].laz += tree[cur].laz;
        tree[cur << 1].val += tree[cur].laz;
        
        tree[cur << 1 | 1].laz += tree[cur].laz;
        tree[cur << 1 | 1].val += tree[cur].laz;
        
        tree[cur].laz = 0;
    }
}
void update(int pl, int pr, int cur, int val) {
    if(pl <= tree[cur].l && tree[cur].r <= pr) {
        tree[cur].val += val;
        tree[cur].laz += val;
        return;
    }
    pushdown(cur);
    if(pl <= tree[cur << 1].r) update(pl, pr, cur << 1, val);
    if(pr >= tree[cur << 1 | 1].l) update(pl, pr, cur << 1 | 1, val);
    pushup(cur);
}
int cnt;
void query(int pl, int pr, int cur) {
    if(tree[cur].l == tree[cur].r) {
        cnt = tree[cur].l;
        return;
    } 
    pushdown(cur);
    if(tree[cur << 1].val >= c) query(pl, pr, cur << 1);
    else if(pr >= tree[cur << 1 | 1].l && tree[cur << 1 | 1].val >= c) query(pl, pr, cur << 1 | 1);
}
int main() {
    int pos;
    int ans;
    int l, r;
    while(~scanf("%d %d %d", &n, &c, &k)) {
        
        for(int i = 1; i <= c; i++) v[i].clear(); 
        
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            v[a[i]].push_back(i);
        }
        ans = 0;
        build(1, n, 1);
        if(k == 1) {
            update(1, 1, 1, 1);
            ans = 1;
        }
        for(int i = 2; i <= n; i++) {
       //     cout << i << ": \n";
            update(i, i, 1, c - 1);
            pos = lower_bound(v[a[i]].begin(), v[a[i]].end(), i) - v[a[i]].begin();
            if(pos + 1 >= k) {
                if(pos == 0) l = 1;
                else l = v[a[i]][pos - 1] + 1;
                r = i - 1;
            //    cout << l << " " << r << endl;
                if(l <= r) update(l, r, 1, -1); // 前面数目为0的区间减去
                
                if(pos - k < 0) l = 1;
                else l = v[a[i]][pos - k] + 1;
                
                r = v[a[i]][pos - k + 1];
            //    cout << l << " " << r << endl;
                if(l <= r) update(l, r, 1, 1); // 数目等于k的区间加上
    
            } else {
            	if(pos == 0) l = 1;
            	else l = v[a[i]][pos - 1] + 1;
            	r = i - 1;
            	if(l <= r) update(l, r, 1, -1);// 前面数目为0的区间减去
			}
            cnt = -1;
            query(1, i, 1);
            if(cnt != -1) ans = max(ans, i - cnt + 1);
        }
        printf("%d\n", ans);
    }
    return 0;
}

分治错代码：（但是过了，数据水），就是按照不符合条件的数来划分区间的，注意这里b是全局变量，在接下来的dfs中就进行更新了，所以是不对的，但是数据水，过了。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n, c, k;
int num[N], a[N], b[N];
int ans, cnt;
void dfs(int l, int r) {
    if(l > r) return;
    cnt = 0;
    for(int i = l; i <= r; i++) num[a[i]] = 0;
    
    for(int i = l; i <= r; i++)     {
        if(num[a[i]] == 0 || num[a[i]] >= k) {
            num[a[i]]++;
            if(num[a[i]] < k)
                cnt++;
        } else {
            num[a[i]]++;
            if(num[a[i]] >= k)
                cnt--;
        }
    }
//    cout << l << " " << r << " " << cnt << endl; 
    if(cnt == 0) {
        ans = max(ans, r - l + 1);
        return;
    }
    int len = 0;
    int pre = l;
    for(int i = l ; i <= r; i ++) {
        
        if(num[a[i]] < k) {
            
            b[++len] = i;
        }
    } 
    b[0] = l - 1;
    b[len + 1] = r + 1;
    for(int i = 0; i <= len; i++) {
        if(b[i] + 1 <= b[i + 1] - 1)
            dfs(b[i] + 1, b[i + 1] - 1);
    }
}
int main() {
    while(~scanf("%d %d %d", &n, &c, &k)) {
        ans = 0;
        for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
        dfs(1, n);
        printf("%d\n", ans);
    }
    return 0;
}