有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。
你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。
请写一个程序,计算你总共能够到达多少块黑色的瓷砖。
输入格式
输入包括多个数据集合。
每个数据集合的第一行是两个整数 WW 和 HH,分别表示 xx 方向和 yy 方向瓷砖的数量。
在接下来的 HH 行中,每行包括 WW 个字符。每个字符表示一块瓷砖的颜色,规则如下
1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。
输出格式
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
数据范围
1≤W,H≤201≤W,H≤20
输入样例:
6 9
....#.
.....#
......
......
......
......
......
#@...#
.#..#.
0 0
输出样例:
45
package main
import (
"fmt"
)
func check( a []byte) (int) {
for i, w := range a {
if w == '@'{
return i
}
}
return -1
}
func bfs(sx , sy , W ,H int , M[][]byte) (int){
var ans int
visited := make(map[[2]int]bool)
q :=make([][2]int,0)
q = append(q, [2]int{sx, sy})
dirs := [][]int{{-1, 0}, {0, -1}, {1, 0}, {0, 1}}
for len(q) != 0 {
node := q[0]
q = q[1:]
if _, ok := visited[node]; ok {
continue
}
ans += 1
for _, dir := range dirs {
x := node[0] + dir[0]
y := node[1] + dir[1]
if x >= 0 && x < H && y >= 0 && y < W && M[x][y] =='.'{
q = append(q, [2]int{x, y})
}
}
visited[node] = true
}
return ans
}
func main () {
var W, H int
loop : fmt.Scanf("%d %d", &W, &H)
// fmt.Println(W,H)
if W != 0 || H!= 0 {
M := make([][]byte, 0)
var sx , sy int
for i:= 0 ; i < H ; i ++ {
var tmp string
fmt.Scanf("%s", &tmp)
// fmt.Println(tmp)
s := []byte(tmp)
// fmt.Println(s)
M = append(M, s)
if check(s) != -1 {
sx = i
sy = check(s)
}
}
// fmt.Println(M)
fmt.Println(bfs(sx,sy, W, H , M))
goto loop
}
}