[关闭]
@jtahstu 2017-09-15T10:07:41.000000Z 字数 1354 阅读 2464

水池数目(深搜)

算法


出处:NYOJ27题

题目

水池数目

时间限制:3000 ms | 内存限制:65535 KB
难度:4

描述

南阳理工学院校园里有一些小河和一些湖泊,现在,我们把它们通一看成水池,假设有一张我们学校的某处的地图,这个地图上仅标识了此处是否是水池,现在,你的任务来了,请用计算机算出该地图中共有几个水池。

输入

第一行输入一个整数N,表示共有N组测试数据
每一组数据都是先输入该地图的行数m(0 < m < 100)与列数n(0 < n < 100),然后,输入接下来的m行每行输入n个数,表示此处有水还是没水(1表示此处是水池,0表示此处是地面)

输出

输出该地图中水池的个数。
要注意,每个水池的旁边(上下左右四个位置)如果还是水池的话的话,它们可以看做是同一个水池。

样例输入

2
3 4
1 0 0 0
0 0 1 1
1 1 1 0
5 5
1 1 1 1 0
0 0 1 0 1
0 0 0 0 0
1 1 1 0 0
0 0 1 1 1

样例输出

2
3

思路

走过的标0,然后深搜

代码

  1. # author: jtusta
  2. # contact: root@jtahstu.com
  3. # site: http://www.jtahstu.com
  4. # software: RubyMine
  5. # time: 2017/6/9 22:08
  6. class PollsNum
  7. def initialize
  8. @a = []
  9. @n = 0
  10. @m = 0
  11. @count = 0
  12. end
  13. def input
  14. t = gets.strip
  15. for x in 1..t.to_i
  16. # 初始化一下
  17. @a = []
  18. @count = 0
  19. # 输入一个二维数组
  20. l = gets.strip.split(/\s+/)
  21. @n = l[0].to_i
  22. @m = l[1].to_i
  23. for i in 1..@n
  24. line = gets.split(/\s+/)
  25. @a[i] = []
  26. for j in 1..@m
  27. @a[i][j] = line[j-1].to_i
  28. end
  29. end
  30. # solve
  31. self.solve
  32. end
  33. end
  34. # 深搜
  35. def dfs(i,j)
  36. if i-1>0 && @a[i-1][j]==1
  37. @a[i-1][j]=0
  38. self.dfs(i-1,j)
  39. end
  40. if i<@n && @a[i+1][j]==1 # 右
  41. @a[i+1][j]=0
  42. self.dfs(i+1,j)
  43. end
  44. if j-1>0 && @a[i][j-1]==1 # 上
  45. @a[i][j-1]=0
  46. self.dfs(i,j-1)
  47. end
  48. if j<@m && @a[i][j+1]==1 # 下
  49. @a[i][j+1]=0
  50. self.dfs(i,j+1)
  51. end
  52. end
  53. def solve
  54. for i in 1..@n
  55. for j in 1..@m
  56. if @a[i][j]==1
  57. @count += 1
  58. @a[i][j]=0
  59. self.dfs(i,j)
  60. end
  61. end
  62. end
  63. p @count
  64. end
  65. end
  66. polls = PollsNum.new
  67. polls.input

输出

  1. /usr/bin/ruby -e $stdout.sync=true;$stderr.sync=true;load($0=ARGV.shift) /Users/jtusta/Documents/Code/Ruby/algorithm/polls_num.rb
  2. 2
  3. 3 4
  4. 1 0 0 0
  5. 0 0 1 1
  6. 1 1 1 0
  7. 2
  8. 5 5
  9. 1 1 1 1 0
  10. 0 0 1 0 1
  11. 0 0 0 0 0
  12. 1 1 1 0 0
  13. 0 0 1 1 1
  14. 3
  15. Process finished with exit code 0
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注