LeetCode-Python-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 条或者更多路径可以进入或离开。
• 保证存在答案

 

思路：

先通过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:]