読者です 読者をやめる 読者になる 読者になる

kanetaiの二次記憶装置

プログラミングに関するやってみた、調べた系のものをQitaに移して、それ以外をはてブでやる運用にしようと思います。http://qiita.com/kanetai

Alien Messages(AOJ No.0236)

リポジトリ

Alien Messages(AOJ No.0236)

W×Hのマップ(0:empty, 1:障害物)が与えられて閉曲線が描けるかどうかを答える。
↓みたいに交差したりしてはいけない。
http://judge.u-aizu.ac.jp/onlinejudge/IMAGE2/UFOMessage1.png

http://judge.u-aizu.ac.jp/onlinejudge/IMAGE2/UFOMessage2.pnghttp://judge.u-aizu.ac.jp/onlinejudge/IMAGE2/UFOMessage3.pnghttp://judge.u-aizu.ac.jp/onlinejudge/IMAGE2/UFOMessage4.png
制約
0 < W, H < 8

アルゴリズム

障害物の無いところに, 左上から、下の6パターン(十字路のわたり方)の線をDFSで書く。パターンの配置は 6^{W*H}=6^{7^2} > 10^{38}あるので枝刈りする。
4近傍を見て明らかに連結できない場合(|_の左側が__の場合など)はバックトラックする。
全部の空きマスに接続パターンを書き込んだら、1つの閉曲線になっているかDFSで調べる。これで一応通る。

  • (x-1, y), (x, y), (x+1, y)
  • (x, y-1), (x, y), (x, y+1)
  • (x, y-1), (x, y), (x+1, y)
  • (x+1, y), (x, y), (x, y+1)
  • (x-1, y), (x, y), (x, y+1)
  • (x-1, y), (x, y), (x, y-1)

コード

接続パターンは4bit(y-1, x-1, y+1, x+1)で覚えてます。

import java.util.*;
public class aoj0236 {
	static final Scanner stdin = new Scanner(System.in);
	static final int N = 7, K = 4;
	static final int[] dx = new int[]{1, 0, -1, 0};
	static final int[] dy = new int[]{0, 1, 0, -1};
	static int W, H, data[][] = new int[N][N];
	static final int PX = 0, PY = 1, NX = 2, NY = 3;
	static final int pattern[] = { bit(NX)|bit(PX), bit(NY)|bit(PY), bit(NY)|bit(PX), bit(PX)|bit(PY), bit(NX)|bit(PY), bit(NX)|bit(NY) };
	static boolean[][] block;
	static boolean inRange(int y, int x) { return 0 <= x && x < W && 0 <= y && y < H; }
	static int bit(int n) { return 1<<n; }
	static boolean bitTrue(int bitset, int n) { return (bitset & bit(n)) != 0; }
	public static void main(String[] args) {
		while (true) {
			W = stdin.nextInt(); H = stdin.nextInt();
			if ((W|H) == 0) break;
			block = new boolean[H][W];
			for (int i = 0; i < H; ++i) for (int j = 0; j < W; ++j) block[i][j] = (stdin.nextInt() == 1);
			System.out.println(rec(0) ? "Yes" : "No");
		}
	}
	static boolean rec(int p) {
	    if(p == W * H) return validate();
	    int x = p % W, y = p / W;
	    if(!block[y][x]) {
	        for (final int pat : pattern) {
	            if(bitTrue(pat, PX) && (x+1 >= W || block[y][x+1])) continue;
	            if(bitTrue(pat, PY) && (y+1 >= H || block[y+1][x])) continue;
	            if(bitTrue(pat, NX) != (x-1 >= 0 && (bitTrue(data[y][x-1], PX)))) continue;
	            if(bitTrue(pat, NY) != (y-1 >= 0 && (bitTrue(data[y-1][x], PY)))) continue;
	            data[y][x] = pat;
	            if(rec(p+1)) return true;
	        }
	    } else {
	        data[y][x] = 0;
	        return rec(p+1);
	    }
	    return false;
	}
	static boolean validate() {
	    boolean ret = false;
	    boolean[][] visited = new boolean[H][W];
	    for (int i = 0; i < H; ++i) for (int j = 0; j < W; ++j) if(!block[i][j] && !visited[i][j]) {
	        if (ret) return false; //multiple lines
	        ret = true;
	        visited[i][j] = true;
	        dfs(i, j, visited);
	    }
	    return ret;
	}
	static void dfs(int y, int x, boolean[][] visited) {
	    for (int k = 0; k < K; ++k) {
	        if(bitTrue(data[y][x], k)) {
	            int nx = x + dx[k], ny = y + dy[k];
	            if(!inRange(ny, nx) || visited[ny][nx]) continue;
	            visited[ny][nx] = true;
	            dfs(ny, nx, visited);
	        }
	    }
	}
}