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];
}
};
舒服多了!
可是 还得加强练啊 😔