Imprecise Computer

题目
https://codeforces.com/gym/102920/problem/E

有两轮操作，每一轮，让$1\sim n$中的每个数字$k$和其他所有数字进行比较大小，如果两个数字之差大于等于$2$则能正确的比较大小，否则随机认为某个数字大，令$r_i(k)$为第$i$轮时比$k$小的数字的个数，则$d_k=|r_1(k)-r_2(k)|$，现在给定$d$数组，问你经过两轮操作后是否可能出现$d$数组这种情况。

思路

可以发现，$d_k$的值只和$k-1$和$k+1$有关，因为$k$和其他数字比较都是确定的，相减贡献就抵消了。令$a_k$表示$k$和$k-1$比较时对差的贡献，$b_k$表示$k$和$k+1$比较时对差的贡献，则$d_k=|a_k+b_k|$，$-1\le a_k,b_k\le 1$，还可以发现$a_k+b_{k+1}=0$，$b_k+a_{k+1}=0$。可以考虑$dp$，令$dp[i][j]$表示考虑前$i$个数字，$b_k=j$时，$d$数组是否符合要求，然后
$$a_k=\{\begin{array}{l}{d}_k-j\\ -d_k-j\end{array}$$
如果$-1\le a_k\le 1$，那么就可以继续算出$b_{k-1}=-a_k$，于是
$$dp[i][j]=dp[i][j]||dp[i-1][b_{k-1}]$$
因为$b_n=0$，所以答案就是$dp[n][0]$。

#include<bits/stdc++.h>
using namespace std;
const int N=1000009;
int n,a[N],dp[N][3],b[3]= {0,1,-1};
map<int,int>ma;
int main() {
    ma[0]=0;
    ma[1]=1;
    ma[-1]=2;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    dp[0][0]=1;
    for(int i=1; i<=n; i++)
        for(int j=0; j<3; j++) {
            int tmp=b[j];
            if(a[i]-tmp>=-1&&a[i]-tmp<=1)
                dp[i][j]=dp[i][j]||dp[i-1][ma[tmp-a[i]]];
            if(-a[i]-tmp>=-1&&-a[i]-tmp<=1)
                dp[i][j]=dp[i][j]||dp[i-1][ma[tmp+a[i]]];
        }
    if(dp[n][0]){
        printf("YES");
        return 0;
    }
    printf("NO");
    return 0;
}