开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第29天，点击查看活动详情
不邻接植花
有 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 <= 104
- 0 <= paths.length <= 2 * 104
- paths[i].length == 2
- 1 <= xi, yi <= n
- xi != yi
- 每个花园 最多 有 3 条路径可以进入或离开
解题思路:
- 找到跟当前花园有路径相连其他的花园 (用哈希map 保存这一结构 key为当前花园 value为数组)
- 当前花园种什么花 取决于 其他相连的花园种的剩下的花; 所以先把其他花园种的花的选项删去,剩下的就是可以种的花
我的答案:
/**
* @param {number} n
* @param {number[][]} paths
* @return {number[]}
*/
var gardenNoAdj = function(n, paths) {
const map = new Map()
const res = new Array(n).fill(1)
// 第一步: 找到跟当前花园有路径相连其他的花园 (用哈希map 保存这一结构 key为当前花园 value为数组)
for (let [key1, key2] of paths) {
map.set(key1, map.get(key1) ? map.get(key1).concat(key2) : [key2])
map.set(key2, map.get(key2) ? map.get(key2).concat(key1) : [key1])
}
for (let [key, value] of map) {
// 第二步: 当前花园种什么花 取决于 其他相连的花园种的剩下的花;
const select = [1, 2, 3, 4]
// 所以先把其他花园种的花的选项删去,剩下的就是可以种的花
for (const garden of value) {
const flower = res[garden-1]
const index = select.indexOf(flower)
select.splice(index, 1)
}
res[key-1] = select[0]
}
return res
};
最后
如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )
