201412-4 最优灌溉

使用kruskal求解,耗时943ms,得分100

徘徊在超时的边缘,同样的代码,有时候提交是100分,有时候是超时90分,还有时候是超时80分==


import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Scanner;

class Node implements Comparable<Node>{
    public int p1;
    public int p2;
    public int l;

    public Node(int p1, int p2, int l) {
        this.p1 = p1;
        this.p2 = p2;
        this.l = l;
    }

    @Override
    public int compareTo(Node o) {
        return this.l - o.l;
    }
}

public class Main{
    private static int[] v;

    public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        String[] firstLine= scanner.nextLine().split(" ");
        int n = Integer.parseInt(firstLine[0]);
        int m = Integer.parseInt(firstLine[1]);
        List<Node> nodes = new ArrayList<>();
        for(int i = 0; i < m; i++){
            String[] line = scanner.nextLine().split(" ");
            int p1 = Integer.parseInt(line[0]);
            int p2 = Integer.parseInt(line[1]);
            int l = Integer.parseInt(line[2]);
            nodes.add(new Node(p1, p2, l));
        }
        scanner.close();

        Collections.sort(nodes);
        v = new int[n + 1];
        for(int i = 0; i < v.length; i++){
            v[i] = i;
        }

        int len = 0;
        while(n > 1){
            Node node = nodes.get(0);
            nodes.remove(node);

            int x = find(node.p1);
            int y = find(node.p2);
            if(x != y){
                len += node.l;
                v[x] = y;
                n--;
            }
        }

        System.out.println(len);
    }

    private static int find(int x){
        while (true){
            if(v[x] == x)
                return x;
            x = v[x];
        }
    }

}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值