不邻接植花
题目描述
有 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;
}
};