水池数目 - Ruby

水池数目

标签（空格分隔）： 算法

---

出处：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，然后深搜

代码

# author: jtusta
# contact: root@jtahstu.com
# site: http://www.jtahstu.com
# software: RubyMine
# time: 2017/6/9 22:08

class PollsNum
  def initialize
    @a = []
    @n = 0
    @m = 0
    @count = 0
  end

  def input
    t = gets.strip
    for x in 1..t.to_i
      # 初始化一下
      @a = []
      @count = 0

      # 输入一个二维数组
      l = gets.strip.split(/\s+/)
      @n = l[0].to_i
      @m = l[1].to_i
      for i in 1..@n
        line = gets.split(/\s+/)
        @a[i] = []
        for j in 1..@m
          @a[i][j] = line[j-1].to_i
        end
      end

      # solve
      self.solve
    end
  end

  # 深搜
  def dfs(i,j)
    if i-1>0 && @a[i-1][j]==1
      @a[i-1][j]=0
      self.dfs(i-1,j)
    end
    if i<@n && @a[i+1][j]==1 # 右
      @a[i+1][j]=0
      self.dfs(i+1,j)
    end
    if j-1>0 && @a[i][j-1]==1 # 上
      @a[i][j-1]=0
      self.dfs(i,j-1)
    end
    if j<@m && @a[i][j+1]==1 # 下
      @a[i][j+1]=0
      self.dfs(i,j+1)
    end
  end

  def solve
    for i in 1..@n
      for j in 1..@m
        if @a[i][j]==1
          @count += 1
          @a[i][j]=0
          self.dfs(i,j)
        end
      end
    end
    p @count
  end
end


polls = PollsNum.new
polls.input

输出

/usr/bin/ruby -e $stdout.sync=true;$stderr.sync=true;load($0=ARGV.shift) /Users/jtusta/Documents/Code/Ruby/algorithm/polls_num.rb
2
3 4
1 0 0 0 
0 0 1 1
1 1 1 0
2
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
3

Process finished with exit code 0