LeeCode刷题-easy-信息传递-cpp

LeeCode刷题-easy-信息传递

首先说说LeeCode OJ的那些坑。。。

1. stdout和输出是不一样的！！！LeeCode的输出，必须是在Solution类中的成员函数中return，这才是OJ中实际的输出。。。（花里胡哨的
2. 在OJ以及生成的模板中写代码，不要随便改代码框架。

题目描述：
小朋友 A 在和 ta 的小伙伴们玩传信息游戏，游戏规则如下：

1. 有 n 名玩家，所有玩家编号分别为 0 ～ n-1，其中小朋友 A 的编号为 0
2. 每个玩家都有固定的若干个可传信息的其他玩家（也可能没有）。传信息的关系是单向的（比如 A 可以向 B 传信息，但 B 不能向 A 传信息）。
3. 每轮信息必须需要传递给另一个人，且信息可重复经过同一个人

给定总玩家数 n，以及按 [玩家编号,对应可传递玩家编号] 关系组成的二维数组 relation。返回信息从小 A (编号 0 ) 经过 k 轮传递到编号为 n-1 的小伙伴处的方案数；若不能到达，返回 0。来源：力扣（LeetCode）
示例 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。来源：力扣（LeetCode）

解题思路：BFS、DFS、DP。
我用的是DFS，BFS做起来可能有些麻烦，DFS更直观一些。DP看到网上有人用，应该也可。

class Solution {
public:
 int sum=0;
int path[10][10]= {0};
int step=0;
    int numWays(int n, vector<vector<int>>& relation, int k) {
        for(auto i:relation)
        {
            path[i[0]][i[1]]=1;
        }
        dfs(0,n, k);
        // cout<<sum;
        return sum;
    }
    void dfs(int start,int n, int k)
    {
        if(step==k)
        {
            if(start==n-1)
                sum++;
            return;
        }
        int i=0;
        for(; i<=n-1; i++)
        {
            if(path[start][i]==1)
            {
                step++;
                dfs(i,n,k);
                step--;
            }
        }
    }
};