結果

問題 No.697 池の数はいくつか
ユーザー takeya_okino
提出日時 2019-07-24 19:56:50
言語 Java
(openjdk 23)
結果
AC  
実行時間 5,735 ms / 6,000 ms
コード長 1,827 bytes
コンパイル時間 2,665 ms
コンパイル使用メモリ 77,860 KB
実行使用メモリ 262,532 KB
最終ジャッジ日時 2024-11-25 15:06:37
合計ジャッジ時間 47,919 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
  static int[] par;
  static int[] rank;
  static int[][] a;
  static int h;
  static int w;
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    h = sc.nextInt();
    w = sc.nextInt();
    par = new int[h * w];
    rank = new int[h * w];
    a = new int[h][w];
    init(h * w);
    for(int i = 0; i < h; i++) {
      for(int j = 0; j < w; j++) {
        a[i][j] = sc.nextInt();
      }
    }
    for(int i = 0; i < h; i++) {
      for(int j = 0; j < w; j++) {
        if(a[i][j] == 1) {
          if((i - 1) >= 0) unite((i * w + j), ((i - 1) * w + j));
          if((i + 1) < h) unite((i * w + j), ((i + 1) * w + j));
          if((j - 1) >= 0) unite((i * w + j), (i * w + j - 1));
          if((j + 1) < w) unite((i * w + j), (i * w + j + 1));
        }
      }
    }
    HashSet<Integer> set = new HashSet<Integer>();
    for(int i = 0; i < (h * w); i++) {
      int i1 = i / w;
      int j1 = i - w * i1;
      if(a[i1][j1] == 1) set.add(find(i));
    }
    System.out.println(set.size());
  }
  static void init(int m) {
      for(int i = 0; i < m; i++) {
        par[i] = i;
        rank[i] = 0;
      }
    }
  static int find(int x) {
      if(par[x] == x) {
        return x;
      } else {
        return par[x] = find(par[x]);
      }
    }
  static void unite(int x, int y) {
      int i1 = x / w;
      int j1 = x - i1 * w;
      int i2 = y / w;
      int j2 = y - i2 * w;
      if(a[i1][j1] == a[i2][j2]) {
      x = find(x);
      y = find(y);
      if(x != y) {
        if(rank[x] < rank[y]) {
          par[x] = y;
        } else {
          par[y] = x;
          if(rank[x] == rank[y]) {
            rank[x]++;
          }
        }
      }
      }
    }
  static boolean same(int x, int y) {
      return find(x) == find(y);
    }
}
0