?1042. 不相邻植花(Python)

更多精彩内容,请关注【力扣简单题】

题目

难度:★★☆☆☆
类型:数组,益智

有 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

如有疑问或建议,欢迎评论区留言~

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • "use strict";function _classCallCheck(e,t){if(!(e instanc...
    久些阅读 2,059评论 0 2
  • 个人学习批处理的初衷来源于实际工作;在某个迭代版本有个BS(安卓手游模拟器)大需求,从而在测试过程中就重复涉及到...
    Luckykailiu阅读 4,792评论 0 11
  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,421评论 0 2
  • 昨天等姐姐上课,在大公园里拍的繁花。 真的是很喜欢这样的一树繁花! 静静的开放,悄悄地生长。 这几日的手帐,慢慢活...
    夏秋爱画阅读 180评论 0 0
  • Chapter1 朋友前段时间在朋友圈高调秀恩爱,我想她终于找到了一个心中的白马王子了,小姑娘特别挑,一直标榜说「...
    橘绿时阅读 496评论 0 2