Codeforces Round #484 (Div. 2) D. Shark

题目链接:D. Shark

题目大意

一个数组a[i], 长度为n，选一个数字k，将所有大于等于k的数字去掉，让剩下的每一段子段长度相等（且字段长度不能为0），求让子段数量最多的最小的k

思路

利用单调栈求出l[i]:=a[i]往左第一个大于a[i]的元素的下标，如果不存在则为0，r[i]:=a[i]往右第一个大于a[i]的元素的下标，如果不存在则为n+1
很容易看出k一定是某个a[i]+1, 所以从大到小遍历a[i], 用一个map<int, int> ms, ms[i] = j记录长度为长度为i的段的数量，最开始只有一个ms[n] = 1, 从大到小遍历a[i]，每次去掉一个a[i], 利用l和r数组更新ms，如果ms里面所有长度的子段都一样，且子段数量增多，则更新k

也可以利用带权并查集，从小到大遍历a[i]，如果a[i]与之前的子段相邻，就将a[i]并到那个子段上去，找到第一个让所有并查集大小相等的a[i], 答案就是a[i]+1

代码

GNU C++17 Accepted 249 ms 5300 KB

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int maxn = 1e5 + 100, inf = 0x3f3f3f3f;
int n;
int a[maxn];
int l[maxn], r[maxn];
map<int, int> mp;
int main()
{
    cin >> n;
    for (int i = 1; i <= n; ++i)
    {
        cin >> a[i];
        mp[a[i]] = i;
    }
    stack<P> s;
    s.push(P(inf, 0));
    for (int i = 1; i <= n; ++i)
    {
        while (s.top().first <= a[i]) s.pop();
        l[i] = s.top().second;
        s.push(P(a[i], i));
    }
    while (!s.empty()) s.pop();
    s.push(P(inf, n + 1));
    for (int i = n; i >= 1; --i)
    {
        while (s.top().first <= a[i]) s.pop();
        r[i] = s.top().second;
        s.push(P(a[i], i));
    }

    sort(a + 1, a + 1 + n, greater<int>());
    map<int, int> ms;
    ms[n] = 1;
    int k = a[1] + 1, cnt = 1;
    for (int i = 1; i <= n; ++i)
    {
        int p = mp[a[i]];
        int len = r[p] - l[p] - 1;
        int len1 = p - l[p] - 1;
        int len2 = r[p] - p - 1;
        ms[len]--;
        if (ms[len] == 0) ms.erase(len);
        if (len1) ms[len1]++;
        if (len2) ms[len2]++;
        if (ms.size() == 1 && (int)ms.begin()->second >= cnt)
        {
            k = a[i + 1] + 1;
            cnt = (int)ms.begin()->second;
        }
    }
    cout << k << endl;

    return 0;
}