Lake Counting(POJ No.2386)

NashSP

已于 2023-03-02 18:14:37 修改

阅读量414

点赞数 1

文章标签： DFS（深度优先搜索）开发语言蓝桥杯

于 2019-02-10 20:55:33 首次发布

本文链接：https://blog.csdn.net/beilunc7/article/details/86929749

版权

有一个大小为 m×n的园子，雨后积起了水。八连通的陆地被认为是连接在一起的。请求出院子里没有被水淹没的陆地个数。（八连通指的是下图相对1的0的部分）

000

010

000

限制条件： n,m<=100

样例输入（1表示陆地，0表示水）

5 5
1 0 0 0 0
0 1 0 0 0
0 0 0 1 0
1 1 0 0 0
0 0 0 0 0

样例输出

思路：从任意的1开始，不停地把邻接的部分用0代替。一次DFS后于初始的这个1连接的所有1（包括它本身）就都被替换成了0，因此直到图中不再存在1为止，总共进行DFS的次数就是答案了。8个方向共对应了8种状态转移，每个格子作为DFS的参数至多被调用一次，所以复杂度为O（8×M×N）=O（M×N）。

Java代码：

import java.util.*;
public class Main{
	static int[][] a;
	static int m,n;
	
	static void dfs(int x,int y){
		a[x][y]=0;
		for(int dx=-1;dx<=1;dx++){
			for(int dy=-1;dy<=1;dy++){
				int nx=x+dx,ny=y+dy;
				if(nx>=0&&nx<m&&ny>=0&&ny<n&&a[nx][ny]==1)
					dfs(nx,ny);
			}
		}
	}
	
	public static void main(String[] args){
		Scanner in=new Scanner(System.in);
		m=in.nextInt();
		n=in.nextInt();
		a=new int[m][n];
		for(int i=0;i<m;i++)
			for(int j=0;j<n;j++)
				a[i][j]=in.nextInt();//1表示陆地，0表示水
		int num=0;
		for(int i=0;i<m;i++)
			for(int j=0;j<n;j++)
				if(a[i][j]==1){
					dfs(i,j);
					num++;
				}
		System.out.println(num);			
	}
}

Python代码：

m,n=map(int,input().split())
lake=[[0]*n]*m
for i in range(m):
    lake[i]=input().split(" ")
    lake[i]=[int(j) for j in lake[i]]
# print(lake)

def dfs(x,y):
    lake[x][y]=0
    for i in range(-1,2):
        for j in range(-1,2):
            nx=x+i
            ny=y+j
            if nx>=0 and ny>=0 and nx<m and ny<n and lake[nx][ny]==1:
                dfs(nx,ny)

count=0  # 陆地个数
for i in range(m):
    for j in range(n):
        if lake[i][j]==1:
            count+=1
            dfs(i,j)

print(count)