[日常训练] 秀姿势

【问题描述】

“蓝猫淘气三千问，看蓝猫，我有姿势我自豪！”话说能考上HYSBZ的孩纸们肯定都是很有姿势的孩纸们，但是大家普遍偏科，都只有一门科目考得好。已知HYSBZ的入学考试科目数量小于等于$10^9$，而有n个学生参加了入学考试。现在HYSBZ要刷人了，招生办每一次刷人会把一个科目考得好的人全部刷掉，但是最多不能刷超过K次。（刷就是不录取）而HYSBZ的校长看录取名单时，最喜欢看的就是连续都是同一个科目考得好的人。他定义完美学生序列为连续且考得好的科目都为同一门的学生序列。现在招生办主任想让你帮他设计一种录取方案，使得最长的完美学生序列尽量长。

【输入格式】sugata.in

共N+1行，第一行2个正整数n，K，n表示入学考试人数，K表示刷人次数上限。
接下来N行，每行仅一个正整数Ai，为i号学生所考得好的科目。

【输出格式】sugata.out

仅1个正整数，为最长的最长完美学生序列。

【样例输入】

9 1
2
7
3
7
7
3
7
5
7

【样例输出】

4

【样例解释】

总共有9个学生，最多只能刷一次学生。
若不刷，最长完美学生序列长度为2
若刷掉考第3门考得好的学生，则学生序列变成2 7 7 7 7 5 7，最长完美学生序列长度为4.

【数据范围与约定】

对于10%的数据：n<=10
对于30%的数据：n<=1000
对于100%的数据：1<=n<=100000

【分析】离散化 + 单调队列

• 先对问题进行转化：我们找到一个区间$[l, r]$，区间内存在不超过$K + 1$种不同的数字，找到出现次数最多的数字（其余数字都可以被直接刷掉），则这种数字的出现次数就可用于更新答案。
• 那么根据数据范围$A_i \le 10^9$，我们就对$A_i$离散化，用数组$cnt$统计当前数字$A_i$出现的次数
• 维护一个单调队列，表示当前的合法区间$[l, r]$，每次将右边界右移，新元素加入队列，利用$cnt$计算区间内不同数字的种数$res$，若超过$K + 1$，则将左边界左移，不断弹出队头的元素，直到$res \le K + 1$。每次对合法区间$[l, r]$统计答案。

【代码】

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 1e5 + 5;
int n, k, Ans, a[N], b[N], cnt[N], h[N];

inline void CkMax(int &x, const int &y) {if (x < y) x = y;}

inline int get()
{
    char ch; int res;
    while ((ch = getchar()) < '0' || ch > '9');
    res = ch - '0';
    while ((ch = getchar()) >= '0' && ch <= '9')
     res = (res << 3) + (res << 1) + ch - '0';
    return res;
}

int main()
{
    freopen("sugata.in", "r", stdin);
    freopen("sugata.out", "w", stdout);
    n = get(); k = get() + 1;
    for (int i = 1; i <= n; ++i) a[i] = b[i] = get();
    sort(b + 1, b + n + 1);
    int m = unique(b + 1, b + n + 1) - b - 1;
    for (int i = 1; i <= n; ++i)
     a[i] = lower_bound(b + 1, b + m + 1, a[i]) - b;
    int t, w, res; t = w = res = 0;
    for (int i = 1; i <= n; ++i)
    {
        if (!cnt[h[++w] = a[i]]++) res++;
        if (res > k) {while (--cnt[h[++t]]); res--;}
        CkMax(Ans, cnt[a[i]]);
        //这里可直接用cnt[a[i]]于更新答案
        //因为若右边界再往右移且数字a[i + 1 ~ r]都与a[i]不相同
        //那么相较于右边界为i的方案，之后的方案都可能是不优的
        //同时我们每次只让右边界右移一位，因此也不会遗漏 
    }
    printf("%d\n", Ans);
    fclose(stdin); fclose(stdout);
    return 0;
}