Problem:
ペンキの色

問題

情報オリンピックの宣伝のために,長方形のベニヤ板にペンキを塗り看板を制作したい.ベニヤ板には色を塗りたくないところにあらかじめ何枚かの長方形のマスキングテープが貼られている.そこでマスキングテープで区切られた領域ごとに別々の色を使いペンキを塗ることにした.例えば,図 5-1 の場合は 5 色のペンキを使う.

入力としてマスキングテープを貼る位置が与えられた時,使うペンキの色の数を求めるプログラムを作成せよ.ただし,ベニヤ板全体がマスキングテープで覆われることはなく,全てのマスキングテープの辺はベニヤ板のいずれかの辺に平行である.

入力

入力は複数のデータセットからなる．各データセットは以下の形式で与えられる．

1 行目にはベニヤ板の幅 w (1 ≤ w ≤ 1000000 となる整数) と高さ h (1 ≤ h ≤ 1000000 となる整数) がこの順に空白区切りで書かれている.

2 行目にはマスキングテープの数 n (1 ≤ n ≤ 1000 となる整数) が書かれている.
続く 3 行目以降の 2 + i 行目 (1 ≤ i ≤ n) には,i 番目に貼るマスキングテープの左下の座標 (x1 , y1 ) と,右上の座標 (x2 , y2 ) が x1 , y1 , x2 , y2 (0 ≤ x1 < x2 ≤ w, 0 ≤ y1 < y2 ≤ h となる整数) の順に空白区切りで書かれている.

ただし,ベニヤ板の左下の角の座標は (0, 0) で右上の角の座標は (w, h) である. 採点用データのうち, 配点の 30% 分は w ≤ 100, h ≤ 100, n ≤ 100 を満たす.

h, w がともに 0 のとき入力の終了を示す. データセットの数は 20 を超えない．

出力

データセットごとに使うペンキの色数を１行に出力する.

入出力例

次の例は図 5-1 の場合である.

入力例

15 6
10
1 4 5 6
2 1 4 5
1 0 5 1
6 1 7 5
7 5 9 6
7 0 9 2
9 1 10 5
11 0 14 1
12 1 13 5
11 5 14 6
0 0

出力例

5

上記問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in JAVA (judge verdict: Memory Limit Exceeded)


import static java.lang.Math.*;
import static java.util.Arrays.*;
import java.util.*;
import java.io.*;

public class Main {
	public static void main(String[] args) {
		new Main().run();
	}

	static void tr(Object... os) {
		System.err.println(deepToString(os));
	}

	Scanner sc = new Scanner(System.in);
	int w, h;
	void run() {
		for (;;) {
			w = sc.nextInt();
			h = sc.nextInt();
			if ((w | h) == 0) break;

			int n = sc.nextInt();

			int[] x = new int[2 * n + 2];
			int[] y = new int[2 * n + 2];
			for (int i = 0; i < n; i++) {
				x[2 * i] = sc.nextInt();
				y[2 * i] = sc.nextInt();
				x[2 * i + 1] = sc.nextInt();
				y[2 * i + 1] = sc.nextInt();
			}
			x[2 * n] = 0;
			x[2 * n + 1] = w;
			y[2 * n] = 0;
			y[2 * n + 1] = h;

			x = normalize(x);
			y = normalize(y);

			short[][] table = new short[2 * n + 10][2 * n + 10];
			short[][] acc = new short[2 * n + 10][2 * n + 10];
			for (int i = 0; i < n; i++) {
				int x1 = x[2 * i + 0];
				int y1 = y[2 * i + 0];
				int x2 = x[2 * i + 1];
				int y2 = y[2 * i + 1];
				table[x1][y1]++;
				table[x1][y2]--;
				table[x2][y1]--;
				table[x2][y2]++;
			}

			acc[0][0] = table[0][0];
			for (int i = 1; i < table.length; i++) {
				acc[i][0] = (short) (acc[i-1][0] + table[i][0]);
				acc[0][i] = (short) (acc[0][i-1] + table[0][i]);
			}
			for (int i = 1; i < table.length; i++) {
				for (int j = 1; j < table.length; j++) {
					acc[i][j] = (short) (table[i][j] + acc[i-1][j] + acc[i][j-1] - acc[i-1][j-1]);
				}
			}

			w = x[2 * n + 1];
			h = y[2 * n + 1];

			int ans = 0;
			for (int i = 0; i < w; i++) {
				for (int j = 0; j < h; j++) {
					if (acc[i][j] == 0) {
						ans++;
						bfs((short)i, (short)j, acc);
					}
				}
			}
			System.out.println(ans);

		}
	}

	void bfs(short x, short y, short[][] acc) {

		short[] queue = new short[2000 * 2000];
		int head = 0;
		int tail = 0;
		queue[tail++] = x;
		queue[tail++] = y;
		acc[x][y] = 2;

		short[] dx = {0, -1, 0, 1};
		short[] dy = {-1, 0, 1, 0};

		while (head < tail) {
			x = queue[head++];
			y = queue[head++];

			for (int i = 0; i < 4; i++) {
				short nx = (short) (x + dx[i]);
				short ny = (short) (y + dy[i]);

				if ((0 <= nx && nx < w && 0 <= ny && ny < h) && acc[nx][ny] == 0) {
					queue[tail++] = nx;
					queue[tail++] = ny;
					acc[nx][ny] = 2;
				}

			}
		}

	}

	int[] normalize(int[] v) {
		int[] res = new int[v.length];
		long[][] t = new long[v.length][2];
		for (int i = 0; i < v.length; i++) {
			t[i][0] = v[i];
			t[i][1] = i;
		}
		Arrays.sort(t, 0, t.length, new Comparator<long[]>(){
			public int compare(long[] a, long[] b){
				if (a[0] != b[0]) return a[0] < b[0] ? -1 : 1;
				return 0;
			}
		});

		int r = 0;
		for (int i = 0; i < v.length; i++) {
			r += (i > 0 && t[i - 1][0] != t[i][0]) ? 1 : 0;
			res[(int)t[i][1]] = r;
		}
		return res;
	}


}