每日一题:LCP 07. 传递信息(DFS)

最新推荐文章于 2024-10-31 10:33:22 发布

软件园里卖煎饼

最新推荐文章于 2024-10-31 10:33:22 发布

阅读量80

点赞数

分类专栏： DFS 文章标签：算法 dfs leetcode

本文链接：https://blog.csdn.net/weixin_45221477/article/details/118380878

版权

DFS 专栏收录该内容

53 篇文章 1 订阅

订阅专栏

解题思路

DFS
1.构造图，n*n的矩阵,连通则为1
2.dfs递归,记录步数还有当前所在位置，再利用当前位置的下标取出road中的数组，如果值为1就代表是连通的
3.如果步数达到要求，且所在位置为n-1则ans++

代码

class Solution {
    int ans = 0;
    public int numWays(int n, int[][] relation, int k) {
        int[][] road = new int[n][n];
        for(int i=0;i<relation.length;i++){
            int[] tem = relation[i];
            road[tem[0]][tem[1]] = 1;
        }
        dfs(0,0,n,k,road);
        return ans;
    }
    
    public void dfs(int index,int step,int n,int k,int[][] road){
        if(step==k){
            if(index==n-1){
                ans++;
            }
            return;
        }
        int[] go = road[index];
        for(int i=0;i<go.length;i++){
            if(go[i]==1){
                dfs(i,step+1,n,k,road);
            }
        }
    }
}