Leetcode LCP 07.传递信息

LCP 07.传递信息

一、题目描述

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

二、思路

在动态规划简单题(?)的专栏，但是我做这道题dp没做出来 😦 太菜了！。
而用dfs写出来了。
dfs：从0号节点开始 深搜，当 当前的步长等于k 并且当前 节点所连接的节点编号为n-1时，那么总路线数+1。否则 从当前节点所连接的节点编号 继续深搜。
相对来说 深搜思路还是比较简单的，但是效率太低。

---

dp没想出来，看一看题解。原来如此！
状态定义：dp[i][j]: 经过 i 轮传递到编号 j 的玩家的方案数.
那么最终求的就是 dp[k][n-1] 经过k轮传到n-1玩家的方案数

经过 i 轮传递到编号 j 的玩家的方案数 就等于 经过 i-1轮 传到 编号 m(可能不止一个,所以遍历吧) 的玩家的方案数 ，<编号m 所连接的节点应该等于j>

边界 dp[0][0]=1 (自己到自己算一种) ，其他 dp[0][j] =0 (从0号开始 0轮传到其他玩家的方案数肯定一个没有)

三、代码实现

dfs：

class Solution {
public:
    int sum=0;
    int numWays(int n, vector<vector<int>>& relation, int k) {
        for(int i=0;i<relation.size();i++){
            if(relation[i][0]==0){
                dfs(i,relation,n,k,0);
            }
        }
        return sum;   
    }
    void  dfs(int i, vector<vector<int>>& relation,int n,int k,int step){
        step++;
        if(step==k){
            if(relation[i][1]==n-1){
            sum++;
            }
        return;
        }
        for(int j=0;j<relation.size();j++)
            if(relation[j][0]==relation[i][1])
                dfs(j,relation,n,k,step);
    }
};

时间效率太低了 ！

---

dp：

class Solution {
public:
    int numWays(int n, vector<vector<int>>& relation, int k) {
        int sum=0;
        int dp[105][105]={0};
        dp[0][0]=1;
        for(int i=1;i<=k;i++){
            for(int j=0;j<n;j++){
                //遍历二维数组,找到和 j 邻接的
                for(int m=0;m<relation.size();m++){
                    if(relation[m][1]==j){
                        dp[i][j]+=dp[i-1][relation[m][0]];
                    }
                }
            }
        }
        return dp[k][n-1];
    }

};

额！也没快多少。看看官方解答。

---

震惊！😦 原来用两层循环就可以了！我也太菜了！

在第二层循环时 直接将 谁连接着谁 给记录下来，为什么自己当时想不到。

class Solution {
public:
    int numWays(int n, vector<vector<int>>& relation, int k) {
        int sum=0;
        int dp[105][105]={0};
        dp[0][0]=1;
        for(int i=1;i<=k;i++){
          for(int j=0;j<relation.size();j++){
              dp[i][relation[j][1]]+=dp[i-1][relation[j][0]];
          }
        }
        return dp[k][n-1];
    }

};

舒服多了！
可是 还得加强练啊 😔