1042. 不邻接植花

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

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/flower-planting-with-no-adjacent
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题意!:
就是有四种颜色的花可以选择,给每个公园种一种颜色的花,给出所有公园之间的路径信息,要求有路径连通的两个公园种的花的颜色不能相同。给出一种满足要求的答案。

思路:
这道题相信对于各位大佬来说是一点难度也没有。就是一个简单的涂色问题,dfs或者bfs都是可以解决的,这是普通的思路。
对于这道题而言,搜索回溯其实都不用,因为题目中给出,保证存在答案,这说明无论我们怎么种花,其实都是可以找到答案的。
因此我们只需要对每一个公园检索一下哪些颜色的花是可以种的从中选一种种就可以了。

代码:

class Solution {
public:
    vector<int> gardenNoAdj(int n, vector<vector<int>>& paths) {
        vector<vector<int>> e(n+1);
        //邻接表存有路径连通的花园
        for(auto &q : paths){
            e[q[0]].push_back(q[1]);
            e[q[1]].push_back(q[0]);
        }
        vector<int> ans(n , 0);
        int color[5] = {0};
        for(int i = 1 ; i <= n ; i++){
            memset(color , 0 , sizeof(color));
            for(int x : e[i]){
                if(ans[x-1] > 0){
                    color[ans[x-1]] = 1;
                }
            }
            for(int j = 1 ; j < 5 ; j++){
                if(!color[j]){
                    ans[i-1] = j;
                    break;
                }
            }
        }
        return ans;
    }
};

你可能感兴趣的:(力扣,算法,深度优先,leetcode)