题解
这题看似最短路径,但是暗藏变化。
首先想到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;
}