【CF1474D】Cleaning

链接

传送门

分析

先分析不交换便可以拿走的情况，发显位于两个端点的石头只能依附于旁边一侧的石头，拿走第一堆，2到n的2变成了第一堆。可以写出判断的函数

int check(int* arr, int end){
    for (int i = 1; i <= end; ++i) {
        arr[i] -= arr[i - 1];
        if (arr[i] < 0) return 0;
    }
    return arr[end] == 0;
}

如果是暴力枚举，然后判断，是一种思路，不过时间复杂度高，我们尝试用差分的思想优化。
pre[i]表示i堆之前的全部拿走，i堆的石头，如果是-1，表示之前会出现不成立的情况。后缀同理，在处理完之后，只需要检查相邻四堆石头即可。

code

注意

1. 在求pre[i]时，如果pre[i - 1] < 0表明前面遇到了不成立的情况，可以直接break
2. 在枚举交换的时候，要注意pre[i - 1] 和 suf[i + 2]取值的合理。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>

using namespace std;
inline int read(){
    int x = 0, op = 1; char ch = getchar();
    while (!isdigit(ch)){
        if (ch == '-') op = -1; ch = getchar();}
    while (isdigit(ch)){
        x = (x << 1) + (x << 3) + (ch ^ 48);
        ch = getchar();
    }
    return x * op;
}

inline void write(int x){
    if (x < 0) putchar('-'), x = - x;
    if (x > 9) write(x / 10);
    putchar('0' + x % 10);
}
const int N = 2e5 + 10;
int pre[N], suf[N], a[N], b[N];

int check(int* arr, int end){
    for (int i = 1; i <= end; ++i) {
        arr[i] -= arr[i - 1];
        if (arr[i] < 0) return 0;
    }
    return arr[end] == 0;
}

void solve(){
    memset(pre, -1, sizeof pre);
    memset(suf, -1, sizeof suf);
    memset(a, 0, sizeof a);
    int n = read();
    for (int i = 1; i <= n; ++i) {
        a[i] = read();
        b[i] = a[i];
    }
    pre[0] = 0;
    for (int i = 1; i <= n; ++i) {
        if (pre[i - 1] < 0) break;
        pre[i] = a[i] - pre[i - 1];
    }
    suf[n + 1] = 0;
    for (int i = n; i >= 1; --i) {
        if (suf[i + 1] < 0) break;
        suf[i] = a[i] - suf[i + 1];
    }

    int flag = check(b, n);
    for (int i = 1; i <= n - 1; ++i) {
        if (flag) break;
        if (pre[i - 1] >= 0 && suf[i + 2] >= 0){
            int temp[] = {0, pre[i - 1], a[i + 1], a[i], suf[i + 2]};
            flag = check(temp, 4);
        }
    }
    printf("%s\n", flag? "YES": "NO");
}

int main(){
    int cases = read();
    while (cases--){
        solve();
    }
}