洛谷 P1162 填涂颜色

原题

Description

由数字 0 组成的方阵中，有一任意形状的由数字 1 构成的闭合圈。现要求把闭合圈内的所有空间都填写成 2。例如：6×6 的方阵（n=6），涂色前和涂色后的方阵如下：

如果从某个 0 出发，只向上下左右 4 个方向移动且仅经过其他 0 的情况下，无法到达方阵的边界，就认为这个 0 在闭合圈内。闭合圈不一定是环形的，可以是任意形状，但保证闭合圈内的 0 是连通的（两两之间可以相互到达）。

0 0 0 0 0 0
0 0 0 1 1 1
0 1 1 0 0 1
1 1 0 0 0 1
1 0 0 1 0 1
1 1 1 1 1 1

Plain text

0 0 0 0 0 0
0 0 0 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 1 2 1
1 1 1 1 1 1

Plain text

Input

每组测试数据第一行一个整数n(1≤n≤30)。

接下来 nn 行，由 00 和 11 组成的n×n 的方阵。

方阵内只有一个闭合圈，圈内至少有一个 00。

Output

已经填好数字 2 的完整方阵。

Sample 1

Input	Output
6 0 0 0 0 0 0 0 0 1 1 1 1 0 1 1 0 0 1 1 1 0 0 0 1 1 0 0 0 0 1 1 1 1 1 1 1	0 0 0 0 0 0 0 0 1 1 1 1 0 1 1 2 2 1 1 1 2 2 2 1 1 2 2 2 2 1 1 1 1 1 1 1

Hint

对于100% 的数据，1≤n≤30。

解题思路

bfs或者dfs思路差不多。

本题难点在于如何确定闭合圈。

我的思路是将闭合圈外的点都标记，这样剩下的点就是闭合圈。

如何标记：首先将所有一都标记，然后将第一行到最后一行，第一列到最后一列开始标记，直到遇到1或者越界停止，这样我们就把所有非闭合圈标记好了，然后根据标记确定照常输出还是改为2。

AC代码

bfs:

#include<iostream>
#include <algorithm>
#include <queue>
#include <utility>
using namespace std;
int a[31][31],n,flag;
int d[][2]={{0,1},{1,0},{-1,0},{0,-1}};
bool vis[31][31];
void bfs(int x,int y){
	queue<pair<int,int>> q;
	vis[x][y]=1;        //最外围都标记为1
	q.push({x,y});
	while(q.size()){ 
		pair<int,int> t=q.front();
		int tx=t.first,ty=t.second;
		q.pop();    
		for(int i=0;i<4;i++){
			int nx=tx+d[i][0],ny=ty+d[i][1];
            //判断是否越界
			if((ny<n-1&&ny>0&&nx<n-1&&nx>0)&&a[nx][ny]==0&&!vis[nx][ny]) {
				vis[nx][ny]=1;
				q.push({nx,ny}) ;
			}
		}
		
	}
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			scanf("%d",a[i]+j);
			if(a[i][j]) vis[i][j]=1;
		}
	} 
	for(int i=0;i<n;i+=n-1)    //第一行、最后一行
	for(int j=0;j<n;j++){
		if(!vis[i][j]) bfs(i,j);
	}
	for(int i=0;i<n;i+=n-1)    //第一列、最后一列
	for(int j=0;j<n;j++){
		if(!vis[j][i]) bfs(j,i);
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
            //判断闭合圈
			if(!vis[i][j]) cout<<"2 ";
			else cout<<a[i][j]<<" ";
		}
		cout<<endl;
	} 
	return 0;
} 

dfs:

#include <iostream>
#include <algorithm>
using namespace std;
int a[32][32],n;
bool vis[32][32];
int d[][2]={{0,1},{1,0},{0,-1},{-1,0}};
void dfs(int x,int y){
	if(x>n+1||x<0||y<0||y>n+1||vis[x][y]) return ;
	vis[x][y]=1;
	for(int i=0;i<4;i++) dfs(x+d[i][0],y+d[i][1]);
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) {
			cin>>a[i][j];  
			if(a[i][j]) vis[i][j]=1;
		}
	}
	dfs(0,0);
	for (int i=1;i<=n;i++){
        for (int j=1;j<=n;j++)
        if (vis[i][j]==0) cout<<2<<' ';
        else cout<<a[i][j]<<' ';
        cout<<'\n';
    }
	return 0;
}