链接
分析
先分析不交换便可以拿走的情况，发显位于两个端点的石头只能依附于旁边一侧的石头，拿走第一堆，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
注意
- 在求pre[i]时，如果pre[i - 1] < 0表明前面遇到了不成立的情况，可以直接break
- 在枚举交换的时候，要注意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();
}
}