class Solution {
public:
int numWays(int n, vector<vector<int>>& relation, int k) {
}
};
LCP 07. 传递信息
小朋友 A 在和 ta 的小伙伴们玩传信息游戏，游戏规则如下：
给定总玩家数 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。
示例 2：
输入：
n = 3, relation = [[0,2],[2,1]], k = 2输出：
0解释：信息不能从小 A 处经过 2 轮传递到编号 2
限制：
2 <= n <= 101 <= k <= 51 <= relation.length <= 90, 且 relation[i].length == 20 <= relation[i][0],relation[i][1] < n 且 relation[i][0] != relation[i][1]原站题解
golang 解法, 执行用时: 0 ms, 内存消耗: 2.1 MB, 提交时间: 2021-06-16 10:31:21
func numWays(n int, relation [][]int, k int) int {
dp := make([][]int, k+1)
for i := 0; i <= k; i++ {
for j := 0; j <= n; j++ {
dp[i] = append(dp[i], 0)
}
}
dp[0][0] = 1
for i := 0; i < k; i++ {
for _, r := range relation {
dp[i+1][r[1]] += dp[i][r[0]]
}
}
return dp[k][n-1]
}
golang 解法, 执行用时: 0 ms, 内存消耗: 2.1 MB, 提交时间: 2021-06-16 10:13:59
func numWays(n int, relation [][]int, k int) (ans int) {
mp := make(map[int][]int)
for _, r := range relation {
mp[r[0]] = append(mp[r[0]], r[1])
}
var dfs func(int, int)
dfs = func(index, num int) {
if num == k {
if index == n - 1 {
ans++
}
return
}
for _, i := range mp[index] {
dfs(i, num+1)
}
}
dfs(0, 0)
return
}