leetcode每日一题——959. 由斜杠划分区域

题目描述：
在由 1 x 1 方格组成的 N x N 网格 grid 中，每个 1 x 1 方块由 /、\ 或空格构成。这些字符会将方块划分为一些共边的区域。（请注意，反斜杠字符是转义的，因此 \ 用 “\” 表示。）。
返回区域的数目。





提示：

• 1 <= grid.length == grid[0].length <= 30
• grid[i][j] 是 ‘/’、’\’、或 ’ '。

解题思路：
这道题又是求连通分量的个数问题。我们可以用并查集解决。
具体代码：

public class 由斜杠划分区域 {
	public static int regionsBySlashes(String[] grid) {
		int N = grid.length;
		int size = 4 * N * N;
		UnionFind unionFind = new UnionFind(size);
		for (int i = 0; i < N; i++) {
			char[] row = grid[i].toCharArray();
			for (int j = 0; j < N; j++) {
				// 二维网格转换为一维表格
				int index = 4 * (i * N + j);
				char c = row[j];
				// 单元格内合并
				if (c == '/') {
					// 合并 0、3，合并 1、2
					unionFind.union(index, index + 3);
					unionFind.union(index + 1, index + 2);
				} else if (c == '\\') {
					// 合并 0、1，合并 2、3
					unionFind.union(index, index + 1);
					unionFind.union(index + 2, index + 3);
				} else {
					unionFind.union(index, index + 1);
					unionFind.union(index + 1, index + 2);
					unionFind.union(index + 2, index + 3);
				}
				// 单元格间合并
				// 向右合并：1（当前）、3（右一列）
				if (j + 1 < N) {
					unionFind.union(index + 1, 4 * (i * N + j + 1) + 3);
				}
				// 向下合并：2（当前）、0（下一行）
				if (i + 1 < N) {
					unionFind.union(index + 2, 4 * ((i + 1) * N + j));
				}
			}
		}
		return unionFind.getCount();
	}

	private static class UnionFind {

		private int[] parent;

		private int count;

		public int getCount() {
			return count;
		}

		public UnionFind(int n) {
			this.count = n;
			this.parent = new int[n];
			for (int i = 0; i < n; i++) {
				parent[i] = i;
			}
		}

		public int find(int x) {
			while (x != parent[x]) {
				parent[x] = parent[parent[x]];
				x = parent[x];
			}
			return x;
		}

		public void union(int x, int y) {
			int rootX = find(x);
			int rootY = find(y);
			if (rootX == rootY) {
				return;
			}

			parent[rootX] = rootY;
			count--;
		}
	}
	//测试
	public static void main(String[] args) {
		String[] s = { "/\\", "\\/" };
		int res = regionsBySlashes(s);
		System.out.println(res);
	}
}

把每一个小正方形分成四个区域，进行合并。