Codeforces D. Cleaning (Round #696 Div.2) (思维 / 枚举)

传送门

题意： 有n堆石子排成一排，每次你可同时在相邻两堆石子中都拿走一颗；试问是否可以通过在最初交换(或不用交换)相邻两堆石子而清扫干净所有石子？

思路：

• 首先考虑 a[1] 是要被消除的，且只能被 a[2] 消除，因此便构成如下的消除次序：a[1]，a[2]-a[1]， a[3]-a[1]+a[1] …
• 考虑次序需满足的条件：任意一个数大于等于0，且最后一个数为0，若不满足则可能需要交换。
• 而对于必须交换的情况来说，即便我们从后往前处理，也会出现一个负数情况；但我们不难发现无论对于正向处理(用p数组存过程)还是反向处理(用b数组存过程)，满足清扫条件的部分其效果都是一样的。
• 即若交换一次 a[i] 和 a[i+1] 即可的情况，相对于p数组和b数组都是首次出现不满足情况的位置；于是需满足 a[i+1] >= p[i-1] && a[i] >= b[i+2] && a[i+1]-p[i-1] == a[i]-b[i+2] 条件即可达到目的。
• 详细思路见大佬博客。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 10;

int t, n;
int a[N], p[N], b[N];

signed main()
{
    IOS;

    cin >> t;
    while(t --){
        cin >> n;
        me(p); me(b);
        for(int i = 1; i <= n; i ++){  //正向处理
            cin >> a[i];
            if(p[i-1]==-1||a[i]<p[i-1]) p[i] = -1;
            else p[i] = a[i]-p[i-1];
        }
        if(!p[n]){ //无需进行交换
            cout << "YES" << endl;
            continue;
        }
        for(int i = n; i; i --){
            if(b[i+1]==-1||a[i]<b[i+1]) b[i] = -1;
            else b[i] = a[i]-b[i+1];
        }
        int flag = 0;
        for(int i = 1; i < n; i ++){  //反向处理
            if(p[i-1]==-1||b[i+2]==-1) continue;
            if(a[i+1]>=p[i-1]&&a[i+1]-p[i-1]==a[i]-b[i+2]){
                flag = 1;
                break;
            }
        }
        cout << (flag ? "YES":"NO") << endl;
    }

    return 0;
}