Leetcode.1042 不邻接植花

题目链接

Leetcode.1042 不邻接植花 Rating ： 1712

题目描述

有 n个花园，按从 1到 n标记。另有数组 paths，其中 paths[i] = [xi, yi]描述了花园 xi到花园 yi的双向路径。在每个花园中，你打算种下四种花之一。

另外，所有花园 最多 有 3条路径可以进入或离开.

你需要为每个花园选择一种花，使得通过路径相连的任何两个花园中的花的种类互不相同。

以数组形式返回 任一 可行的方案作为答案 answer，其中 answer[i]为在第 (i+1) 个花园中种植的花的种类。花的种类用 1、2、3、4 表示。保证存在答案。

示例 1：

输入：n = 3, paths = [[1,2],[2,3],[3,1]]
输出：[1,2,3]
解释：
花园 1 和 2 花的种类不同。
花园 2 和 3 花的种类不同。
花园 3 和 1 花的种类不同。
因此，[1,2,3] 是一个满足题意的答案。其他满足题意的答案有 [1,2,4]、[1,4,2] 和 [3,2,1]

示例 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<=10^4$
• $0<=paths.length<=2\ast 10^4$
• $paths[i].length==2$
• $1<=xi,yi<=n$
• $xi!=yi$
• 每个花园 最多 有 3条路径可以进入或离开

分析：

对于一个点 u，我们可以先记录它的邻接点 v种了哪些花，u再从剩下的花中选一种来种植。

对于每一个点，它的路径最多只有 3条，所以我们可以直接遍历。

时间复杂度：$O(3\ast n)$

C++代码：

class Solution {
public:
    vector<int> gardenNoAdj(int n, vector<vector<int>>& paths) {
        //邻接表
        vector<vector<int>> g(n,vector<int>());
        for(auto &e:paths){
            //因为是 1~n 我们把它映射成 0~n-1 方便处理
            int a = e[0] - 1 , b = e[1] - 1;
            g[a].push_back(b);
            g[b].push_back(a);
        }

        vector<int> ans(n);

        for(int i = 0;i < n;i++){
            
            int color[5] = {0};
            
            //先记录 i 的邻接点 v 都选了哪些花来种
            for(auto v:g[i]){
                color[ans[v]] = 1;
            }
            
            //color[c] == 0 的就是剩下的花，再选择它来种在 i 这个点上
            for(int c = 1;c <= 4;c++){
                if(color[c] == 0){
                    ans[i] = c;
                    break;
                }
            }
        }

        return ans;
    }
};

Java代码：

class Solution {
    public int[] gardenNoAdj(int n, int[][] paths) {
        List<List<Integer>> g = new ArrayList<>();
        for(int i = 0;i < n;i++) g.add(new ArrayList<>());

        for(int[] e:paths){
            int a = e[0] - 1;
            int b = e[1] - 1;
            g.get(a).add(b);
            g.get(b).add(a);
        }

        int[] res = new int[n];

        for(int i = 0;i < n;i++){
            int[] color = new int[]{0,0,0,0,0};
            for(int v:g.get(i)){
                color[res[v]] = 1;
            }

            for(int c = 1;c <= 4;c++){
                if(color[c] == 0){
                    res[i] = c;
                    break;
                }
            }
        }

        return res;
    }
}