题解 | #继续畅通工程#注意如果已经建好的话要把花费设为0

继续畅通工程

https://www.nowcoder.com/practice/16212f7d46e44174b5505997ea998538

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;
public class Main {
    private static int[] parents;
    private static class Edge {
        private final int from, to, weight;
        private Edge(int from, int to, int weight) {
            this.from = from;
            this.to = to;
            this.weight = weight;
        }
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while (sc.hasNextInt()) {
            int n = sc.nextInt();
            if (n == 0) break;
            List<Edge> edges = new ArrayList<>();
            for (int i = 0; i < n * (n - 1) / 2; i++) {
                int from = sc.nextInt();
                int to = sc.nextInt();
                int weight = sc.nextInt();
                int flag = sc.nextInt();
                if (flag == 1) weight = 0;
                edges.add(new Edge(from, to, weight));
            }
            edges.sort(Comparator.comparingInt(o -> o.weight));
            parents = new int[n + 1];
            for (int i = 1; i <= n; i++) parents[i] = i;
            int ans = 0;
            for (Edge edge : edges) {
                int a = find(edge.from);
                int b = find(edge.to);
                if (a != b) {
                    parents[a] = b;
                    ans += edge.weight;
                }
            }
            System.out.println(ans);
        }
    }
    private static int find(int x) {
        return x == parents[x] ? x : (parents[x] = find(parents[x]));
    }
}

全部评论

相关推荐

不懂!!!:感觉你的项目描述太简单了,建议使用star描述法优化提炼一下,就是使用什么技术或方案解决了什么问题,有什么效果或成果,例如:对axios进行了二次封装,实现了请求的统一管理、错误的集中处理以及接口调用的简化,显著提高了开发效率和代码维护性,使用canvas技术实现了路线绘制功能,通过定义路径绘制函数和动态更新机制,满足了简化的导航可视化需求,提升了用户体验。像什么是使用其他组件库,基本功能描述就最好不要写到项目成果里面去了,加油
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务