結果

問題 No.307 最近色塗る問題多くない？
ユーザー uafr_csuafr_cs
提出日時 2015-11-27 23:59:09
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 4,200 bytes
コンパイル時間 2,750 ms
コンパイル使用メモリ 80,116 KB
実行使用メモリ 80,788 KB
最終ジャッジ日時 2024-09-14 01:42:04
合計ジャッジ時間 10,983 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 140 ms
53,900 KB
testcase_01 AC 141 ms
54,176 KB
testcase_02 AC 146 ms
53,916 KB
testcase_03 AC 141 ms
53,904 KB
testcase_04 WA -
testcase_05 AC 209 ms
54,364 KB
testcase_06 AC 245 ms
56,228 KB
testcase_07 TLE -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Set;

public class Main {
	
	public static final int[][] nexts = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
	public static boolean is_in(int x, int y, int W, int H){
		return 0 <= x && x < W && 0 <= y && y < H;
	}
	
	public static class UnionFind{
	    int[] par; // 
	    
	    public UnionFind(int n){
	        par = new int[n];
	        for(int i = 0; i < n; i++){ par[i] = -1; }
	    }
	    
	    public int find(int x){
	        if(par[x] < 0){
	            return x;
	        }else{
	            return par[x] = find(par[x]);
	        }
	    }
	    
	    public boolean union(int x, int y){
	        x = find(x);
	        y = find(y);
	        
	        if(x != y){
	            if(par[y] < par[x]) {  // 多い方が根になるようにスワップする.
	                int tmp = x; x = y; y = tmp;
	            }
	            par[x] += par[y];
	            par[y] = x;
	            return true;
	        }else{
	            return false;
	        }
	    }
	    
	    public boolean same(int x, int y){
	        return find(x) == find(y);
	    }
	    
	    public int size(int x){
	        return -par[find(x)];
	    }
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		final int H = sc.nextInt();
		final int W = sc.nextInt();
		
		boolean[][] colors = new boolean[H][W];
		for(int i = 0; i < H; i++){
			for(int j = 0; j < W; j++){
				colors[i][j] = sc.nextInt() == 1;
			}
		}
		
		UnionFind uf = new UnionFind(H * W);
		ArrayList<HashSet<Integer>> adjs = new ArrayList<HashSet<Integer>>(H * W);
		for(int y = 0; y < H; y++){
			for(int x = 0; x < W; x++){
				final HashSet<Integer> set = new HashSet<Integer>();
				
				final int id = y * W + x;
				final int uf_id = uf.find(id);
				
				for(int[] next : nexts){
					final int nx = x + next[0];
					final int ny = y + next[1];
					
					if(!is_in(nx, ny, W, H)){
						continue;
					}else if(colors[ny][nx] == colors[y][x]){
						continue;
					}
					
					final int next_id = ny * W + nx;
					set.add(next_id);
				}
					
				adjs.add(set);
			}	
		}
		
		for(int y = 0; y < H; y++){
			for(int x = 0; x < W; x++){
				final int id = y * W + x;
				final int uf_id = uf.find(id);
				
				for(int[] next : nexts){
					final int nx = x + next[0];
					final int ny = y + next[1];
					
					if(!is_in(nx, ny, W, H)){
						continue;
					}
					
					final int next_id = ny * W + nx;
					final int next_uf_id = uf.find(next_id);
					
					if(colors[next_uf_id / W][next_uf_id % W] == colors[uf_id / W][uf_id % W]){
						HashSet<Integer> merged = new HashSet<Integer>();
						merged.addAll(adjs.get(uf_id));
						merged.addAll(adjs.get(next_uf_id));
						merged.remove(uf_id);
						merged.remove(next_uf_id);
						
						uf.union(uf_id, next_uf_id);
						
						adjs.set(uf_id, merged);
						adjs.set(next_uf_id, merged);
					}
				}
				
			}
		}
		
		final int Q = sc.nextInt();
		for(int i = 0; i < Q; i++){
			final int y = sc.nextInt() - 1;
			final int x = sc.nextInt() - 1;
			final boolean color = sc.nextInt() == 1;
			
			final int id = y * W + x;
			final int uf_id = uf.find(id);
			
			if(colors[uf_id / W][uf_id % W] == color){
				continue;
			}
			colors[uf_id / W][uf_id % W] = color;
			
			//System.out.println(i + " orig : " + (uf_id / W) + " " + (uf_id % W));
			for(final int next_uf_id : adjs.get(uf_id)){
				//System.out.println(i + " : " + (next_uf_id / W) + " " + (next_uf_id % W));
				HashSet<Integer> merged = new HashSet<Integer>();
				merged.addAll(adjs.get(uf_id));
				merged.addAll(adjs.get(next_uf_id));
				merged.remove(uf_id);
				merged.remove(next_uf_id);
				colors[next_uf_id / W][next_uf_id % W] = color;
				
				uf.union(uf_id, next_uf_id);
				
				adjs.set(uf_id, merged);
				adjs.set(next_uf_id, merged);
			}
		}
		
		
		for(int y = 0; y < H; y++){
			for(int x = 0; x < W; x++){
				final int uf_id = uf.find(y * W + x);
				if(x != 0){
					System.out.printf(" ");
				}
				System.out.print(colors[uf_id / W][uf_id % W] ? 1 : 0);
			}
			System.out.println();
		}
		
	}

}
0