1042. 不邻接植花
有 N 个花园,按从 1 到 N 标记。在每个花园中,你打算种下四种花之一。
paths[i] = [x, y] 描述了花园 x 到花园 y 的双向路径。
另外,没有花园有 3 条以上的路径可以进入或者离开。
你需要为每个花园选择一种花,使得通过路径相连的任何两个花园中的花的种类互不相同。
以数组形式返回选择的方案作为答案 answer,其中 answer[i] 为在第 (i+1) 个花园中种植的花的种类。花的种类用 1, 2, 3, 4 表示。保证存在答案。
示例 1:
输入:N = 3, paths = [[1,2],[2,3],[3,1]]
输出:[1,2,3]
示例 2:
输入:N = 4, paths = [[1,2],[3,4]]
输出:[1,2,1,2]
示例 3:
输入:N = 4, paths = [[1,2],[2,3],[3,4],[4,1],[1,3],[2,4]]
输出:[1,2,3,4]
提示:
1 <= N <= 10000
0 <= paths.size <= 20000
不存在花园有 4 条或者更多路径可以进入或离开。
保证存在答案。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flower-planting-with-no-adjacent
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
注意是一个点不能与之相连的点相同颜色(图着色问题),而不是整条链路需要不同
func gardenNoAdj(N int, paths [][]int) []int {
graph := make([][]int, N)
for _, path := range paths {
i, j := path[0] - 1, path[1] - 1
if graph[i] == nil {
graph[i] = make([]int, 0)
}
if graph[j] == nil {
graph[j] = make([]int, 0)
}
graph[i] = append(graph[i], j)
graph[j] = append(graph[j], i)
}
res := make([]int, N)
for src, dsts := range graph {
visits := make([]bool, 4)
for _, dst := range dsts {
if res[dst] != 0 {
visits[res[dst] - 1] = true
}
}
for i, visit := range visits {
if visit {
continue
}
res[src] = i + 1
break
}
}
return res
}