题目:
有 N 个花园,按从 1 到 N 标记。在每个花园中,你打算种下四种花之一。
paths[i] = [x, y] 描述了花园 x 到花园 y 的双向路径。
另外,没有花园有 3 条以上的路径可以进入或者离开。
你需要为每个花园选择一种花,使得通过路径相连的任何两个花园中的花的种类互不相同。
以数组形式返回选择的方案作为答案 answer,其中 answer[i] 为在第 (i+1) 个花园中种植的花的种类。花的种类用 1, 2, 3, 4 表示。保证存在答案。
示例
提示:
解题思路:
首先,使用邻接矩阵的话,亲测,会堆栈溢出,遂改为邻接表法。
1、根据paths建立邻接表;
2、默认所有的花园先不染色,即染0;
3、从第一个花园开始走,把与它邻接的花园的颜色从color{1,2,3,4}这个颜色集中删除;
4、删完了所有与它相邻的颜色,就可以把集合中剩下的颜色随机选一个给它了,为了简单,将集合中的第一个颜色赋给当前花园;
5、循环3和4到最后一个花园。
class Solution {
public int[] gardenNoAdj(int N, int[][] paths) {
/* 这是一道简单题,限制每个节点的度为3,同时提供四种颜色,因此不需要回溯 */
/* 初始化节点,使用map保存节点与其临界点的关系 */
/* 第一版本采用了内部类构建,参考评论区的HashMap更简洁 */
Map<Integer, Set<Integer>> graph = new HashMap<>();
for (int i = 0; i < N; i++) {
graph.put(i, new HashSet<>());
}
/* 初始化路径信息 */
for (int[] path: paths) {
int a = path[0] - 1;
int b = path[1] - 1;
graph.get(a).add(b);
graph.get(b).add(a);
}
int[] res = new int[N];
for (int i = 0; i < N; i++) {
boolean[] used = new boolean[5];
/* 查看当前节点的所有邻接点的色彩 */
for (int adj: graph.get(i)) {
used[res[adj]] = true;
}
/* 为当前节点染色 */
for (int j = 1; j <= 4; j++) {
if (!used[j]) {
res[i] = j;
}
}
}
return res;
}
}
分析:
这是一道简单题,限制每个节点的度为3,同时提供四种颜色,因此不需要回溯
存储邻接点信息
遍历所有节点,对于每个节点
查看其邻接点颜色,使用不同的颜色染色即可