LeetCode1042. 不邻接植花

不邻接植花

题目描述

有 N 个花园，按从 1 到 N 标记。在每个花园中，你打算种下四种花之一。

paths[i] = [x, y] 描述了花园 x 到花园 y 的双向路径。

另外，没有花园有 3 条以上的路径可以进入或者离开。

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

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

解题思路

有n个花园，4种花，相邻花园之间不能种相同的花
一个一个花园开始种花，然后种和相邻花园不同的花，依次往下种

代码

class Solution {
public:
    vector<int> gardenNoAdj(int N, vector<vector<int>>& paths) {
        vector<vector<int>>vec(N);//无向邻接表
        vector<int>ans(N,0);//最后结果
        for(int i=0;i<paths.size();i++)
        {
            vec[paths[i][0]-1].push_back(paths[i][1]);
            vec[paths[i][1]-1].push_back(paths[i][0]);
        }
        for(int i=0;i<N;i++)//按花园依次种花
        {
            int m[4]={0};//四种花的种植情况
            for(int j=0;j<vec[i].size();j++)//有几个花园与其相邻
            {
                if(ans[vec[i][j]-1]!=0)//如果与该花园相邻的花园已经种植了花，将该种花记录
                {
                    m[ans[vec[i][j]-1]-1]++;
                }
            }
            for(int k=0;k<4;k++)//看四种花的种植情况
            {
                if(m[k]==0)//如果k+1没被种
                {
                    ans[i]=k+1;//第i个花园种植k+1
                    break;
                }
            }
        }
        return ans;
    }
};