Leetcode上LCP 07.传递信息

Leetcode上LCP 07.传递信息

我的思路答案：

// 小朋友 A 在和 ta 的小伙伴们玩传信息游戏，游戏规则如下：
// 有 n 名玩家，所有玩家编号分别为 0 ～ n-1，其中小朋友 A 的编号为 0
// 每个玩家都有固定的若干个可传信息的其他玩家（也可能没有）。传信息的关系是单向的（比如 A 可以向 B 传信息，但 B 不能向 A 传信息）。
// 每轮信息必须需要传递给另一个人，且信息可重复经过同一个人
// 给定总玩家数 n，以及按 [玩家编号,对应可传递玩家编号] 关系组成的二维数组 relation。返回信息从小 A (编号 0 ) 经过 k 轮传递到编号为 n-1 的小伙伴处的方案数；若不能到达，返回 0。

// 示例 1：
// 输入：n = 5, relation = [[0,2],[2,1],[3,4],[2,3],[1,4],[2,0],[0,4]], k = 3
// 输出：3
// 解释：信息从小 A 编号 0 处开始，经 3 轮传递，到达编号 4。共有 3 种方案，分别是 0->2->0->4， 0->2->1->4， 0->2->3->4。

function find(fi, di, mp, count,line,result) {
    let arr = mp[fi];
    if (arr === undefined){
        return 0;
    }

    if (count === 0) {
        if (arr.indexOf(di) !== -1) {
            let road = [...line,di];
            // console.log(JSON.stringify(road));
            result.push(road);
            return 1;
        } else {
            return 0;
        }
    }

    count--;
    // console.log("fi = ",fi)


    for (let i = 0; i < arr.length; i++) {
        let k = arr[i];
        let road = [...line,k];
        //console.log(road);
        find(k, di, mp, count,road,result);
    }
    return 0;
}

function deal(arr, n, k) {
    let mp = {};
    for (let i = 0; i < arr.length; i++) {
        let [k, v] = arr[i];
        if (mp[k] === undefined) {
            mp[k] = []
        }

        mp[k].push(v);
    }

    let result =[];
    let line = [];
    find(0, n - 1, mp, k-1,line,result);
    // console.log(result.length);
    return result.length;
}


/**
 * @param {number} n
 * @param {number[][]} relation
 * @param {number} k
 * @return {number}
 */
var numWays = function(n, relation, k) {
    return deal(relation,n,k)
};

let c1 = numWays(5,
[[0,2],[2,1],[3,4],[2,3],[1,4],[2,0],[0,4]],
3);
console.log('c1 ',c1)

let c2 = numWays(5,
[[1,4],[1,3],[2,3],[4,0],[0,1],[0,3],[2,4],[0,4],[4,3],[1,2]],
3);
console.log(c2)

网上标准答案：

// 小朋友 A 在和 ta 的小伙伴们玩传信息游戏，游戏规则如下：
// 有 n 名玩家，所有玩家编号分别为 0 ～ n-1，其中小朋友 A 的编号为 0
// 每个玩家都有固定的若干个可传信息的其他玩家（也可能没有）。传信息的关系是单向的（比如 A 可以向 B 传信息，但 B 不能向 A 传信息）。
// 每轮信息必须需要传递给另一个人，且信息可重复经过同一个人
// 给定总玩家数 n，以及按 [玩家编号,对应可传递玩家编号] 关系组成的二维数组 relation。返回信息从小 A (编号 0 ) 经过 k 轮传递到编号为 n-1 的小伙伴处的方案数；若不能到达，返回 0。

// 示例 1：
// 输入：n = 5, relation = [[0,2],[2,1],[3,4],[2,3],[1,4],[2,0],[0,4]], k = 3
// 输出：3
// 解释：信息从小 A 编号 0 处开始，经 3 轮传递，到达编号 4。共有 3 种方案，分别是 0->2->0->4， 0->2->1->4， 0->2->3->4。

/**
 * @param {number} n
 * @param {number[][]} relation
 * @param {number} k
 * @return {number}
 */
var numWays = function (n, relation, k) {
    const G = new Array(n)
    for (let i = 0; i < n; i++) {
        G[i] = []
    }
    for (const [x, y] of relation) {
        G[x].push(y)
    }
    let dp = new Array(n).fill(0)
    dp[0] = 1
    for (let i = 0; i < k; i++) {
        const curr = new Array(n).fill(0)
        for (let j = 0; j < n; j ++) {
            for (const v of G[j]) {
                curr[v] += dp[j]
            }
        }
        dp = curr
    }
    return dp[n - 1]
};

let c1 = numWays(5,
[[0,2],[2,1],[3,4],[2,3],[1,4],[2,0],[0,4]],
3);
console.log('c1 ',c1)

let c2 = numWays(5,
[[1,4],[1,3],[2,3],[4,0],[0,1],[0,3],[2,4],[0,4],[4,3],[1,2]],
3);
console.log('c2 ', c2)