2065. 最大化一张图中的路径价值

发布于 2024-07-01  5 次阅读


给你一张 无向 图,图中有 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);
                }
            }
        }
    }
}