leet-code/src/main/java/leetcode/editor/cn/MaximumPathQualityOfAGraph.java

156 lines
5.0 KiB
Java
Raw Normal View History

//给你一张 无向 图,图中有 n 个节点,节点编号从 0 到 n - 1 (都包括)。同时给你一个下标从 0 开始的整数数组 values ,其中
//values[i] 是第 i 个节点的 价值 。同时给你一个下标从 0 开始的二维整数数组 edges ,其中 edges[j] = [uj, vj, timej] 表示
//节点 uj 和 vj 之间有一条需要 timej 秒才能通过的无向边。最后,给你一个整数 maxTime 。
//
// 合法路径 指的是图中任意一条从节点 0 开始,最终回到节点 0 ,且花费的总时间 不超过 maxTime 秒的一条路径。你可以访问一个节点任意次。一条合法
//路径的 价值 定义为路径中 不同节点 的价值 之和 (每个节点的价值 至多 算入价值总和中一次)。
//
// 请你返回一条合法路径的 最大 价值。
//
// 注意:每个节点 至多 有 四条 边与之相连。
//
//
//
// 示例 1
//
//
//
//
//输入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 。
//
//
// 示例 2
//
//
//
//
//输入values = [5,10,15,20], edges = [[0,1,10],[1,2,10],[0,3,10]], maxTime = 30
//输出25
//解释:
//一条可能的路径为0 -> 3 -> 0 。总花费时间为 10 + 10 = 20 <= 30 。
//访问过的节点为 0 和 3 ,最大路径价值为 5 + 20 = 25 。
//
//
// 示例 3
//
//
//
//
//输入values = [1,2,3,4], edges = [[0,1,10],[1,2,11],[2,3,12],[1,3,13]], maxTime
//= 50
//输出7
//解释:
//一条可能的路径为0 -> 1 -> 3 -> 1 -> 0 。总花费时间为 10 + 13 + 13 + 10 = 46 <= 50 。
//访问过的节点为 0 1 和 3 ,最大路径价值为 1 + 2 + 4 = 7 。
//
// 示例 4
//
//
//
//
//输入values = [0,1,2], edges = [[1,2,10]], maxTime = 10
//输出0
//解释:
//唯一一条路径为 0 。总花费时间为 0 。
//唯一访问过的节点为 0 ,最大路径价值为 0 。
//
//
//
//
// 提示:
//
//
// n == values.length
// 1 <= n <= 1000
// 0 <= values[i] <= 10⁸
// 0 <= edges.length <= 2000
// edges[j].length == 3
// 0 <= uj < vj <= n - 1
// 10 <= timej, maxTime <= 100
// [uj, vj] 所有节点对 互不相同 。
// 每个节点 至多有四条 边。
// 图可能不连通。
//
// 👍 1 👎 0
package leetcode.editor.cn;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
//5921:最大化一张图中的路径价值
class MaximumPathQualityOfAGraph {
public static void main(String[] args) {
//测试代码
Solution solution = new MaximumPathQualityOfAGraph().new Solution();
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int maximalPathQuality(int[] values, int[][] edges, int maxTime) {
Map<Integer, List<int[]>> map = new HashMap<>();
for (int i = 0; i < values.length; i++) {
map.put(i, new ArrayList<>());
}
for (int[] edge : edges) {
List<int[]> list = map.get(edge[0]);
list.add(new int[]{edge[1], edge[2]});
map.put(edge[0], list);
list = map.get(edge[1]);
list.add(new int[]{edge[0], edge[2]});
map.put(edge[1], list);
}
max = values[0];
int[] use = new int[values.length];
use[0] = 1;
pathDfs(map, values[0], use, maxTime, 0, 0, values);
return max;
}
int max;
private void pathDfs(Map<Integer, List<int[]>> map, int value, int[] use, int maxTime, int time, int index, int[] values) {
List<int[]> list = map.get(index);
for (int[] arr : list) {
time += arr[1];
if (time > maxTime) {
time -= arr[1];
continue;
}
if (arr[0] > 0 && use[arr[0]] == 2) {
continue;
}
if (use[arr[0]] == 0) {
value += values[arr[0]];
}
use[arr[0]]++;
if (arr[0] == 0) {
max = Math.max(value, max);
}
if (time == maxTime) {
time -= arr[1];
if (use[arr[0]] == 1) {
value -= values[arr[0]];
}
use[arr[0]]--;
continue;
}
pathDfs(map, value, use, maxTime, time, arr[0], values);
time -= arr[1];
if (use[arr[0]] == 1) {
value -= values[arr[0]];
}
use[arr[0]]--;
}
}
}
//leetcode submit region end(Prohibit modification and deletion)
}