N. Best Solution Unknown（线段树+思维）

2020-2021 Winter Petrozavodsk Camp, Belarusian SU Contest (XXI Open Cup, Grand Prix of Belarus) N. Best Solution Unknown

题目链接

题意： 现在有 $n$ 个数，每一轮可以进行的操作：取相邻的两个数进行比较，较大的获胜(若两数相同，双方都可能获胜)，将较小的去除，并且较大的那个数 $+1$ 。 问 $n-1$ 次操作后，哪些位置上的数留到了最后？

思路：

错误的做法：一开始，我先用线段树从左到右预处理了该位置 $i$ 要吃掉前面所有数的下限，从右往左同理，然后根据极限情况判断是否该位置是否合法。但是忘记考虑可以反复横跳了…直接 $wa9$

正解：对于一段区间，区间的最大值肯定能吃掉其他所有的数，因此，该最大值只要满足该区间的下限就能符合题目要求。然后分为两个两个区间递归下去。最大值的序号用线段树求一下就行。

接下来来讲下限的设置：初始情况 $[1,n]$ 下限是 $0$ 。然后找到该区间的最大值的位置 $m$，分为 $[1,m-1]$ 和 $[m+1,r]$ 两个区间，这两个区间的下限都是 $a[m]$ ，因为，左边的半个区间的经过若干步骤得到的一个数如果大于等于 $a[m]$ ，那么就能继续吃掉 $a[m]$ ，进而吃掉右半部分区间，反过来同理。

那么现在对于下限为 $C$ 区间 $[L,R]$ ，找到最大值的位置 $M$，此时若 $a[M]+R-L$ （即最大值的位置吸收掉该区间所有数） 大于等于下限 $C$ ，那么位置 $M$ 就满足条件。继续递归下去，分成 $[L,M-1]$ 和 $[M+1,R]$ 。这个下限的设置，以左半区间为例，设 $i\in [L,M-1]$ ，位置 $i$ 的数要吸收到左半区间所有的数，然后继续吸收 $a[M]$ ，那么新下限 $C^{\prime }\ge a[M]$ ，但同时要满足，吸收完左半区间，再吸收完整个区间 $[L,R]$ 时的下限，即 $C^{\prime }+R-M+1\ge C$ ，因此，左半区间的下限为 C ′ = m a x { a [ M ] , C + M − R − 1 } C'=max\{a[M],C+M-R-1\} C′=max{a[M],C+M−R−1}。 右半区间同理。

代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<cmath>
#include<map>
#include<set>
#include<cstring>
#include<string>
#include<algorithm>
#define fi first
#define se second
//#include<stdlib.h>
//#include <time.h>
//srand((unsigned)time(NULL));
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x3f3f3f3f;
using namespace std;
const int N = 2e6 + 10;
int n;
int a[N];

#define ls (rt << 1)
#define rs ((rt << 1) | 1)
int tr[N << 2];
int MAX(int x, int y) {
    if (a[x] >= a[y]) return x;
    else return y;
}

void build(int l, int r, int rt) {
    tr[rt] = 0;
    if (l == r) {
        tr[rt] = l;
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, ls);
    build(mid + 1, r, rs);
    tr[rt] = MAX(tr[ls], tr[rs]);
}

int query(int s, int t, int l, int r, int rt) {
    if (s <= l && r <= t) return tr[rt];
    int ans = 0;
    int mid = (l + r) >> 1;
    if (s <= mid) ans = query(s, t, l, mid, ls);
    if (mid < t) ans = MAX(ans, query(s, t, mid + 1, r, rs));
    return ans;
}

int ok[N];
void dfs(int l, int r, int c) {
    if (l > r) return;
    if (l == r) {
        if (a[l] >= c) ok[l] = 1;
        return;
    }
    int mx_id = query(l, r, 1, n, 1);
//    cout << mx_id << "\n";
    if ((a[mx_id] + r - l) >= c) {
        ok[mx_id] = 1;
        int cc = max(a[mx_id], c + mx_id - r - 1);
        dfs(l, mx_id - 1, cc);
        cc = max(a[mx_id], c + l - mx_id - 1);
        dfs(mx_id + 1, r, cc);
    }
}

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]), ok[i] = 0;
    if (n == 1) {
        printf("1\n");
        printf("1\n");
        return 0;
    }
    build(1, n, 1);
    dfs(1, n, 0);
    int cnt = 0;
    for (int i = 1; i <= n; i++)
        if (ok[i]) cnt++;
    printf("%d\n", cnt);
    for (int i = 1; i <= n; i++)
        if (ok[i]) printf("%d ", i);
    printf("\n");

    return 0;
}