题目描述
给你一张 无向 图,图中有 n 个节点,节点编号从 0 到 n - 1 (都包括)。同时给你一个下标从 0 开始的整数数组 values ,其中 values[i] 是第 i 个节点的 价值 。同时给你一个下标从 0 开始的二维整数数组 edges ,其中 edges[j] = [uj, vj, timej] 表示节点 uj 和 vj 之间有一条需要 timej 秒才能通过的无向边。最后,给你一个整数 maxTime 。
合法路径 指的是图中任意一条从节点 0 开始,最终回到节点 0 ,且花费的总时间 不超过 maxTime 秒的一条路径。你可以访问一个节点任意次。一条合法路径的 价值 定义为路径中 不同节点 的价值 之和 (每个节点的价值 至多 算入价值总和中一次)。
请你返回一条合法路径的 最大 价值。
注意:每个节点 至多 有 四条 边与之相连。
输入:values = [0,32,10,43], edges = [[0,1,10],[1,2,15],[0,3,10]], maxTime = 49
输出:75
解释:
一条可能的路径为:0 -> 1 -> 0 -> 3 -> 0 。总花费时间为 10 + 10 + 10 + 10 = 40 <= 49 。
访问过的节点为 0 ,1 和 3 ,最大路径价值为 0 + 32 + 43 = 75 。
思路分析
- 关于原路返回问题: 仅仅与最前面的
判断return语句
是否对vis数组进行判断来决定,所以这里不必有这个条件判断 - 涉及到回溯问题:即可能的不同路径之间可能有交叉
- 周赛没时间看,如果放在第一题,还是有希望写完的额
注意点
- dfs参数最好不要有vis数组,因为会拷贝构造会浪费大量时间复杂度和空间复杂度,这也是当时最后几个测试用例超时的原因
Source code
class Solution {
public:
vector<vector<pair<int,int>>> g;
int maxValue=0;
int maximalPathQuality(vector<int>& values, vector<vector<int>>& edges, int maxTime) {
vector<int> t(values.size(),0);
//1.store the map
function<void(int,int,int)> dfs=[&](int i,int val,int len){ //传入下一层的val和t(隔离不同的路径)
if(len>maxTime) return;
if(i==0&&maxValue<val)
maxValue=val;
for(auto &s:g[i]){
if(t[s.first]==0){
t[s.first]=1;
dfs(s.first,val+values[s.first],len+s.second);
t[s.first]=0;
}
else dfs(s.first,val,len+s.second);
}
};
g.resize(values.size());
for(auto &t:edges){
g[t[0]].emplace_back(t[1],t[2]);
g[t[1]].emplace_back(t[0],t[2]);
}
maxValue=values[0];
t[0]=1;
//2.iteration step
dfs(0,maxValue,0);
return maxValue;
}
};