[leetcode]1786. 从第一个节点出发到最后一个节点的受限路径数(Dijkstra+记忆化搜索/dp)

题目链接

题意

给定一个无向连通图,edges={u,v,w} 表示 u u u和 v v v之间有一条无向边,边权为 w w w
n n n个点 [ 1 , n ] [1,n] [1,n] 每个点到 n n n的最短路为 d i s [ i ] dis[i] dis[i]

定义受限路径: 从起点 1 1 1到 n n n,路径上的 d i s [ i ] dis[i] dis[i]递减

1->n的受限路径方案数

方法一 Dijkstra+记忆化搜索

思路

通过Dijkstra预处理出每个点距离 n n n的最短路

通过dfs枚举每种方案 加上记忆化 避免超时

Code

cpp 复制代码
using ll = long long;
#define pii pair<int,int>
#define ar2 array<ll,2>
void cmax(int &a,int b){a=max(a,b);};
void cmin(int &a,int b){a=min(a,b);};
const int N=1e5+10,MOD=1e9+7,INF=0x3f3f3f3f;const long long LINF=LLONG_MAX;const double eps=1e-6;
int a[N];
class Solution {
public:
    int countRestrictedPaths(int n, vector<vector<int>>& edges) {
        vector<vector<ar2>>g(n+1);
        for(auto e:edges){
            int u=e[0],v=e[1],w=e[2];
            g[u].push_back({w,v});
            g[v].push_back({w,u});
        }

        vector<ll>dis(n+1,LINF);//注意要用LINF,INF太小了
        dis[n]=0;
        priority_queue<pii,vector<pii>,greater<>>pq;
        pq.emplace(0,n);

        while(pq.size()){
            auto [d,i]=pq.top();pq.pop();
            if(d>dis[i]) continue;

            for(auto [w,j]:g[i]){
                if(dis[j]>d+w){
                    dis[j]=d+w;
                    pq.emplace(dis[j],j);
                }
            }
        }
        // if(dis[1]==LINF) return 0;
        for(int i=1;i<=n;i++){
            if(dis[i]==LINF) dis[i]=-1;
        }

        cout<<dis[1]<<endl;

        vector<ll>memo(n+1,-1);
        function<ll(int)>dfs=[&](int i)->ll{
            if(i==n) return 1;
            if(memo[i]!=-1) return memo[i];

            int ret=0;
            for(auto [w,j]:g[i]){
                if(dis[i]>dis[j]){
                    ret=(ret+dfs(j))%MOD;
                }
            }
            // printf("dfs(%d),retrun %d\n",i,ret%MOD);
            memo[i]=ret%MOD;
            return memo[i];
        };
        
        return dfs(1)%MOD;

    }
};

实现细节

  • 注意 Dijkstra要从终点 n n n开始扩展最短路,因为 d i s [ i ] dis[i] dis[i]在此题中表示 i i i到 n n n的距离,跟常见的Dijkstra不同 不是表示 i i i到 1 1 1的距离
    记得开longlong,并且无穷大也要用longlong的max来表示,不然有一个样例过不去

方法二 Dijkstra+dp

思路

思路同上,把记忆化搜索改成dp,我感觉没有记忆化搜索直观,但是能快一点点

Code

cpp 复制代码
using ll = long long;
#define pii pair<int,int>
#define ar2 array<ll,2>
#define endl '\n'
void cmax(int &a,int b){a=max(a,b);};
void cmin(int &a,int b){a=min(a,b);};
const int N=1e5+10,MOD=1e9+7,INF=0x3f3f3f3f;const long long LINF=LLONG_MAX;const double eps=1e-6;

class Solution {
public:
    int countRestrictedPaths(int n, vector<vector<int>>& edges) {
        vector<vector<ar2>>g(n+1);
        for(auto e:edges){
            int u=e[0],v=e[1],w=e[2];
            g[u].push_back({w,v});
            g[v].push_back({w,u});
        }

        vector<ll>dis(n+1,LINF);
        dis[n]=0;
        priority_queue<pii,vector<pii>,greater<>>pq;
        pq.emplace(0,n);

        while(pq.size()){
            auto [d,i]=pq.top();pq.pop();
            if(d>dis[i]) continue;

            for(auto [w,j]:g[i]){
                if(dis[j]>d+w){
                    dis[j]=d+w;
                    pq.emplace(dis[j],j);
                }
            }
        }
        // if(dis[1]==LINF) return 0;
        for(int i=1;i<=n;i++){
            if(dis[i]==LINF) dis[i]=-1;
        }

        vector<int>idx(n);
        iota(idx.begin(),idx.end(),1);
        sort(idx.begin(),idx.end(),[&](int a,int b){
            return dis[a]<dis[b];
        });//按最短路从小到大排序

        vector<ll>dp(n+1);//dp[i]表示从i到n的方案数
        dp[n]=1;
        for(int i:idx){
            for(auto [_,j]:g[i]){
                if(dis[i]>dis[j]){
                    dp[i]=(dp[i]+dp[j])%MOD;
                }
            }
        }
        
        return dp[1];
    }
};

实现细节

根据最短路从小到大更新dp

因为受限路径要求dp[i]要根据dis小于dis[i]的值更新

相关推荐
竹子_2315 分钟前
二分法详解:用生活例子 + 图示
python·算法·生活
用户4822137167751 小时前
Python——正则表达式(字符匹配-贪婪模式)
算法
二向箔reverse1 小时前
经典算法题解析:从思路到实现,掌握核心编程思维
算法
天一生水water1 小时前
聚类的肘部法则原理
算法·支持向量机·聚类
Lenyiin2 小时前
《LeetCode 热题 100》整整 100 题量大管饱题解套餐 中
java·c++·python·leetcode·面试·刷题·lenyiin
蒟蒻小袁2 小时前
力扣面试150题--颠倒二进制位
java·算法·leetcode
boyedu2 小时前
比特币区块链共识算法:数字世界的信任密码
算法·区块链·共识算法·数字货币·加密货币·虚拟货币
pk_xz1234562 小时前
社区资源媒体管理系统设计与实现
网络·python·深度学习·算法·数据挖掘·媒体
2 小时前
LeetCode Hot 100 括号生成
算法·leetcode·职场和发展
逝雪Yuki2 小时前
Leetcode——42. 接雨水
c++·算法·leetcode·双指针·接雨水