正难则反,倒序操作,操作的一定是四个同色格子(或者其中有已经完成操作的点,其他同色),可以从此入手进行 DFS。
遍历每一个点,对其进行 DFS,首先检查是否已经完成处理(被标记为 0)、越界或者不符合条件(有杂颜色),如果没有问题,标记这四个格子(改为 0),对其四周的格子进行搜索,具体的,分别是以下 8 个点:
1 2 3 4
| dfs(x-1,y);dfs(x-1,y+1); dfs(x+2,y);dfs(x+2,y+1); dfs(x,y-1);dfs(x+1,y-1); dfs(x,y+2);dfs(x+1,y+2);
|
于是给出代码,注意倒序输出:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
| #include<bits/stdc++.h> using namespace std; int n,m; int a[1001][1001]; vector<int>ansX; vector<int>ansY; vector<int>ansC; void dfs(int x,int y){ if(x>=n||y>=m||x<1||y<1||a[x][y]+a[x+1][y]+a[x][y+1]+a[x+1][y+1]==0){ return; } int val=max({a[x][y],a[x+1][y],a[x][y+1],a[x+1][y+1]}); if((a[x][y]&&val!=a[x][y])||(a[x+1][y]&&val!=a[x+1][y])||(a[x][y+1]&&val!=a[x][y+1])||(a[x+1][y+1]&&val!=a[x+1][y+1])){ return; } a[x][y]=a[x+1][y]=a[x][y+1]=a[x+1][y+1]=0; ansX.push_back(x); ansY.push_back(y); ansC.push_back(val);
dfs(x-1,y);dfs(x-1,y+1); dfs(x+2,y);dfs(x+2,y+1); dfs(x,y-1);dfs(x+1,y-1); dfs(x,y+2);dfs(x+1,y+2); } int main(){ cin>>n>>m; for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ cin>>a[i][j]; } } for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ dfs(i,j); } } for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ if(a[i][j]!=0){ cout<<-1<<endl; return 0; } } } cout<<ansX.size()<<endl; for(int i=ansX.size();i>=1;i--){ cout<<ansX[i-1]<<" "<<ansY[i-1]<<" "<<ansC[i-1]<<endl; } return 0; }
|