更多精彩内容，请关注【力扣简单题】。
题目
难度：★★☆☆☆
类型：数组，益智
有 N 个花园，按从 1 到 N 标记。在每个花园中，你打算种下四种花之一。
paths[i] = [x, y] 描述了花园 x 到花园 y 的双向路径。
另外，没有花园有 3 条以上的路径可以进入或者离开。
你需要为每个花园选择一种花，使得通过路径相连的任何两个花园中的花的种类互不相同。
以数组形式返回选择的方案作为答案 answer，其中 answer[i] 为在第 (i+1) 个花园中种植的花的种类。花的种类用 1, 2, 3, 4 表示。保证存在答案。
提示
1 <= N <= 10000
0 <= paths.size <= 20000
不存在花园有 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]
解答
参考官网解答。
class Solution:
def gardenNoAdj(self, N, paths):
from collections import defaultdict
garden_flower = {} # 记录已经种过花的花园钟的是哪一种
path_dict = defaultdict(list) # 记录各个花园相连的花园，list的长度不会超过3
for path in paths:
path_dict[path[0]].append(path[1])
path_dict[path[1]].append(path[0])
flower = set([1, 2, 3, 4]) # 一共4种花
result = []
# 从第一个花园开始，依次种花，遍历相连的花园都种了啥，然后看看这个花园可以种啥
for i in range(1, N+1):
if i in path_dict:
used_flower = []
for connected_garden in path_dict[i]:
if connected_garden in garden_flower:
used_flower.append(garden_flower[connected_garden])
this_garden_flower = (flower - set(used_flower)).pop()
else:
this_garden_flower = 1
result.append(this_garden_flower)
garden_flower[i] = this_garden_flower
return result
如有疑问或建议，欢迎评论区留言~
