LeetCode 787. Cheapest Flights Within K Stops

题解

这题看似最短路径，但是暗藏变化。
首先想到dfs，搜索到所有由始至终的可行路径，记录最小cost即可。
但是这题编程实现有很多小花招，怎么剪枝，一个是k(步进的次数)，
二是cost，都可以用来减小搜索空间。

再者，就是贝尔福特算法，非常适合这题，因为此题有个k的限制。
当然应用的时候需要修改一点，原本是重复|V|-1次的，此时只要k+1次了。
值得注意的是，在每一轮更新dis[]数组的时候，我们不能直接更新。
相反，我们要用一个tmp[]记录此轮更新。结束时，再使得dis=tmp。
为什么？因为原本的做法会得到全局最优，但这种方法不考虑步进数。
比如，我们限定了k=0，即只能在一步之遥内考虑最短距离。
原本的方法，可能在边集合遍历的某种很好的顺序下恰好一次就走遍了全部点。
而这，与我们k的限制相悖，故必须只能用tmp记录一次的变化。
最后，ans遍历所有的tmp[dst]取最小值。

---

Code

bellman-ford

int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        int INF = 1000001;
        vector<int> dist(n,INF);
        dist[src]=0;
        int ans=INF;
        for(int i=0;i<=K;i++){
            vector<int> tmp(n,INF);// 步进式
            for(auto p:flights){
                if(dist[p[1]] > dist[p[0]]+p[2])
                    dist[p[1]]=dist[p[0]]+p[2];
            }
            dist = tmp;// 步进式
            ans = min( ans, dist[dst]);
        }
        return ans ==INF ? -1 : ans ;
    }

dfs

void dfs(int k,int cost,int &ans,vector<bool> &vis,vector<vector<int>> &cot, int src,int dst){
        if(src==dst){
            ans=cost;
            return;
        }
        if(k<=0) return;// 超过 最大边次数
        
        vis[src]=true;
        for(int i=0;i<cot.size();i++){
        // note！-> cot[src][i]+cost < ans
            if(cot[src][i]>0 && cot[src][i]+cost < ans && !vis[i]){
                dfs(k-1,cost+cot[src][i],ans,vis,cot,i,dst);
            }
        }
        vis[src]=false;
    }
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        vector<vector<int>> cot(n,vector<int>(n,-1));
        for(auto lk:flights){
            cot[lk[0]][lk[1]] = lk[2];
        }
        vector<int> res;
        vector<bool> vis(n,false);
        int ans = 1000001;
        dfs(K+1,0,ans,vis,cot,src,dst);        
        return ans==1000001?-1:ans;
    }