有 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 条或者更多路径可以进入或离开。
- 保证存在答案
思路:
先通过paths把建立一个dictionary,key是每个花园,val是跟key相连的花园。
然后再处理每个key,把它和它的相连的花园不冲突地安排好,
因为答案一定存在,所以顺序处理下去就好了。
注意花的种类只有四种,一开始以为很多种所以疯狂超时……
class Solution(object):
def gardenNoAdj(self, N, paths):
"""
:type N: int
:type paths: List[List[int]]
:rtype: List[int]
"""
#每个花园最多和3个别的花园相连
# if not paths:
# return []
from collections import defaultdict
dic = defaultdict(list)
for path in paths:
x0, y0 = path[0], path[1]
dic[x0].append(y0)
dic[y0].append(x0)
# print dic
res = [-1 for _ in range(N + 1)]
# print len(dic.keys())
for i in dic.keys():
neibors = dic[i]
used = [0 for _ in range(5)]
for neibor in neibors:
if res[neibor] != -1:
used[res[neibor]] = 1
if sum(used) == 0:
res[i] = 1
else:
for j in range(1, 5):
if used[j] == 0:
res[i] = j
break
# print res
for i, x in enumerate(res):
if x == -1:
res[i] = 1
return res[1:]