原题链接:Leetcode 1042. 不邻接植花
class Solution {
public:
vector<int> gardenNoAdj(int n, vector<vector<int>>& paths) {
vector<int> res(n+1);
vector<vector<int>> adj(n+1);
for(auto x:paths)
{
int a=x[0],b=x[1];
adj[a].push_back(b);
adj[b].push_back(a);
}
for(int i=1;i<=n;i++)
{
set<int> color={1,2,3,4};
for(auto x:adj[i]) color.erase(res[x]);
res[i]=*(color.begin());
}
res.assign(res.begin()+1,res.end());
return res;
}
};