最优灌溉(堆优化的最小生成树alg)

18 篇文章 0 订阅
5 篇文章 0 订阅
时间限制:1.0s
内存限制:256.0MB
问题描述:

问题描述

  雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。
  为了灌溉,雷雷需要建立一些水渠,以连接水井和麦田,雷雷也可以利用部分麦田作为“中转站”,利用水渠连接不同的麦田,这样只要一片麦田能被灌溉,则与其连接的麦田也能被灌溉。
  现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用(注意不是所有麦田之间都可以建立水渠)。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

  输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
  接下来m行,每行包含三个整数ai, bi, ci,表示第ai片麦田与第bi片麦田之间可以建立一条水渠,所需要的费用为ci。

输出格式

  输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

  建立以下三条水渠:麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

  前20%的评测用例满足:n≤5。
  前40%的评测用例满足:n≤20。
  前60%的评测用例满足:n≤100。
  所有评测用例都满足:1≤n≤1000,1≤m≤100,000,1≤ci≤10,000。

堆优化克鲁斯卡尔算法,纯的不能再纯了。。。。因为用py写耗费了大量时间了解排序函数,以后当板子用了


class node:
    fro=to=dist=0
def getfx(x):
    if fa[x]==x:
        return fa[x]
    fa[x] = getfx(fa[x])
    return fa[x]
def merge(x,y):
    tx = getfx(x)
    ty = getfx(y)
    fa[tx] = ty    
while True:
    try:
        graph = []
        n,m = map(int,input().split())
        for i in range(m):
            a,b,c = map(int,input().split())
            graph.append([a,b,c])
            #print(graph[i].dist)
        graph = sorted(graph,key = lambda x:x[2])
        # for i in range(m):
            # print(graph[i][2])
        fa = [0 for i in range(1005)]
        for i in range(n+1):
            fa[i]=i
        #print(1)
        # for i in range(m):
            # print(graph[i].dist)
        ans = 0
        last = n
        k = 0
        while k<m and last>1:
            u = graph[k][0]
            v = graph[k][1]
            if not (getfx(u)==getfx(v)):
                merge(u,v)
                last-=1
                ans+=graph[k][2]
            k+=1                
        print(ans)    
    except:             
        break


 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值