Leetcode---LCP 07传递信息

最新推荐文章于 2023-01-12 00:56:32 发布

譕訫_

最新推荐文章于 2023-01-12 00:56:32 发布

阅读量223

点赞数

CC 4.0 BY-SA版权

分类专栏： Java学习 Leetcode每日一题文章标签： java leetcode

本文链接：https://blog.youkuaiyun.com/weixin_41963310/article/details/118382238

Java学习同时被 2 个专栏收录

88 篇文章

订阅专栏

Leetcode每日一题

72 篇文章

订阅专栏

本文介绍了一种利用广度优先搜索(BFS)算法解决特定路径计数问题的方法。通过维护两个队列分别记录节点及对应的步数，该算法可以在O(nk)的时间复杂度内找到从起点出发到达终点的所有可能路径中，恰好经过k步的路径数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

思路

使用广度优先搜索BFS进行处理，因为在计算过程中可能会设计多个元素重复出现，使用Map存储不合适，使用另一个队列存储对应元素的对应step。
因为最终求的是step等于k的方案数量，所以当出现第一个step大于k的元素时证明后续的元素已经不存在可行的方案了，所以直接可以break。
时间复杂度为 $O(n^k)$

代码

class Solution {
    public int numWays(int n, int[][] relation, int k) {
        int ans = numWaysBFS(relation, n, k);
        return ans;
    }
    public int numWaysBFS(int[][] r,int num,int k)
    {
        Deque<Integer> d = new ArrayDeque<>();//存储元素
        Deque<Integer> d2 = new ArrayDeque<>();//存储元素对应的step
        d.addLast(0);
        d2.addLast(0);
        int count = 0;
        while( !d.isEmpty())
        {
            int p = d.pollFirst();

            int step = d2.pollFirst();
//            System.out.println(p+"--"+step);
            if (step>k)
                break;
            if (step==k && p==(num-1))
                count++;
            for (int i=0;i<r.length;i++)
            {
                if (r[i][0]==p)
                {
                    d2.addLast(step+1);
                    d.addLast(r[i][1]);
                }
            }

        }

        return count;
    }
}