每日一题力扣 1377 https://leetcode.cn/problems/frog-position-after-t-seconds/-526互联

力扣 1377 https://leetcode.cn/problems/frog-position-after-t-seconds/

这道题目用dp去做,构建邻接矩阵，做的时候需要注意题目条件，如果青蛙跳不动了，这个概率就保持不变了

一般跳青蛙，很容易想到dp

核心代码如下

public double frogPosition(i

public double frogPosition(int n, int[][] edges, int t, int target) {


//    用邻接矩阵表示
    int[][] graph = new int[n+1][n+1];


    for (int i = 0; i < edges.length; i++) {
        int[] edge = edges[i];

        int start=edge[0];
        int end=edge[1];
    //    无向图
        graph[start][end]=1;
        graph[end][start]=1;

    }
    int[] isVisit = new int[n+1];
    //1.dp[i][j] 表示i顶点 j秒时候的概率
    double[][] dp = new double[n+1][t+1];


//    2.递推见循环

//    3.初始化1号点
    dp[1][0]=1;
    isVisit[1]=1;
    for (int time = 1; time <= t; time++) {

    //    拿出当前time--的值 从它们出发 ，然后计算能够到达的点

        for (int i = 0; i < dp.length; i++) {

            if (dp[i][time-1]!=0){
            //    这个地方可以往下面走

            //    顶点是i

            //    找顶点i的临界节点
              List<Integer> neithbor =getNeighbor(graph,i,isVisit);

            //  这些节点 新加概率
                for (int j = 0; j < neithbor.size(); j++) {

                    dp[neithbor.get(j)][time]+=dp[i][time-1]/neithbor.size();
                    isVisit[neithbor.get(j)]=1;

                }


            //    如果neithbor.size是0的话，说明谁也到不了，以后这个点的概率就固定死了
                if (neithbor.isEmpty()){
                    for (int j = time; j <= t; j++) {
                        dp[i][j]=dp[i][time-1];
                    }
                }


            }

        }


    }





    // for (int i = dp[target].length-1; i >=0; i--) {
    //     if (dp[target][t]!=0){
    //         return dp[target][t];
    //     }
    //     t--;
    // }

    return dp[target][t];







}

private List<Integer> getNeighbor(int[][] graph, int start, int[] isVisit) {

    ArrayList<Integer> list = new ArrayList<>();
    int[] neithbor = graph[start];


    for (int i = 0; i < neithbor.length; i++) {
        if (neithbor[i]==1){
            if (isVisit[i]==0){
                list.add(i);
            }
        }
    }
    return list;
}