题解 | 明日DISCO

如果所有数均相等，那么只能等于边界0。

#include <iostream>
#include<vector>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    cin>>n;
    vector<vector<int>>a(n+2,vector<int>(n+2,0));
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>a[i][j];
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(a[i][j]==0)continue;
            if(a[i-1][j]>=0&&a[i+1][j]>=0&&a[i][j-1]>=0&&a[i][j+1]>=0&&a[i][j]<0){
                a[i][j]=0;
            }
            if(a[i-1][j]<=0&&a[i+1][j]<=0&&a[i][j-1]<=0&&a[i][j+1]<=0&&a[i][j]>0){
                a[i][j]=0;
            }
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(a[i][j]){
                cout<<"NO\n";
                return 0;
            }
        }
    }
    cout<<"YES\n";
    return 0;
}