题目解析
本题其实就是求解连通图的染色方案,
目前我想到的最好方式是暴力法,即通过回溯算法,求解出染红节点的全组合,
n个数的全组合数量一共有 (2^n) - 1。
比如:1,2,3的全组合情况有:1、2、3、12、13、23、123,即 (2^3) - 1 = 7个。
本题中节点一共有m个,而1 <= m <= 15,即最多有 (2^15) - 1 = 32767 个组合情况,这个数量级不算多。 因此暴力法可行。
我们需要尝试对组合中的节点进行染红色,但是相邻节点不能同时染成红色。因此,在求解全组合时,还可以进行剪枝优化,即判断新加入的节点 是否和 已存在的节点相邻,如果相邻,则剪枝,如果不相邻则继续递归。
// 连通图的染色方案数求解
function getDyeCount(arr, m) {
// connect用于存放每个节点的相邻节点
const connect = {};
for (let [v1, v2] of arr) {
connect[v1] ? connect[v1].add(v2) : (connect[v1] = new Set([v2]));
connect[v2] ? connect[v2].add(v1) : (connect[v2] = new Set([v1]));
}
// 必有一种全黑的染色方案
let count = 1;
// 求解染红节点的全组合情