给你一张 无向 图,图中有 n
个节点,节点编号从 0
到 n - 1
(都包括)。同时给你一个下标从 0 开始的整数数组 values
,其中 values[i]
是第 i
个节点的 价值 。同时给你一个下标从 0 开始的二维整数数组 edges
,其中 edges[j] = [uj, vj, timej]
表示节点 uj
和 vj
之间有一条需要 timej
秒才能通过的无向边。最后,给你一个整数 maxTime
。
合法路径 指的是图中任意一条从节点 0
开始,最终回到节点 0
,且花费的总时间 不超过 maxTime
秒的一条路径。你可以访问一个节点任意次。一条合法路径的 价值 定义为路径中 不同节点 的价值 之和 (每个节点的价值 至多 算入价值总和中一次)。
请你返回一条合法路径的 最大 价值。
注意:每个节点 至多 有 四条 边与之相连。
class Solution {
int ans = 0;
int[] values;
int maxTime;
List<int[]>[] g;
boolean[] visited;
public int maximalPathQuality(int[] values, int[][] edges, int maxTime) {
this.values = values;
this.maxTime = maxTime;
int n = values.length;
g = new List[n];
for(int i = 0; i < n; i++){
g[i] = new ArrayList<int[]>();
}
for(int[] edge : edges){
g[edge[0]].add(new int[]{edge[1],edge[2]});
g[edge[1]].add(new int[]{edge[0],edge[2]});
}
visited = new boolean[n];
visited[0] = true;
dfs(0, 0, values[0]);
return ans;
}
public void dfs(int u, int time, int value){
if(u == 0){
ans = Math.max(ans, value);
}
for(int[] arr : g[u]){
int v = arr[0],dist = arr[1];
if(time + dist <= maxTime){
if(!visited[v]){
visited[v] = true;
dfs(v, time + dist, value + values[v]);
visited[v] = false;
}else {
dfs(v, time + dist, value);
}
}
}
}
}
Comments NOTHING