这个题着实花了一些功夫,因为自己对于这种最短距离或者最优路线之类的题不敏感,而且一直没有理解动态规划的解法,理解不了就无从下手,这里记录一下自己所经受的折磨,也算是总结一下。
先将程序贴在这:
class Solution {
public:
int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
//dp[i][k]表示从src至多经过k站到达i的最少费用
vector<vector<int>> dp(n, vector<int>(K + 1, 100000));
//初始化 src 到 src的费用为0
for (int k = 0; k <= K; ++k){
dp[src][k] = 0;
}
//初始化从src不经过中转站到达各地的最少花费
for(auto &flight:flights){
if(flight[0] == src)
dp[flight[1]][0] = flight[2];
}
//开始动态规划
for (int k = 1; k < K + 1; ++k){
for (auto &flight : flights){