round e

本文探讨了一种特殊的樱桃连接问题,通过构建最小生成树模型,使用黑色和红色糖线连接N个樱桃,计算最终的最小含糖量。采用Python实现算法,巧妙地利用并查集数据结构确定最小生成树中各边的颜色,从而得到最优解。
摘要由CSDN通过智能技术生成

共有N个樱桃,目前N个樱桃中有M条黑色糖线(黑色表示含糖量为1个单位),为使所有樱桃连在一起(最小生成树的感觉),可能还需红色糖线(红色含糖量为2个单位),问最终最少的含糖量。(原链接:https://codingcompetitions.withgoogle.com/kickstart/round/0000000000050edb/0000000000170721)

	
import numpy as np

def find(x):
    if f[x]!=x:
        f[x]=find(f[x])   # 寻找祖先节点
    return f[x]

case=int(input())
for k in range(1,case+1):
    N,M=[int(s) for s in input().split(" ")]
    ans=(N-1)*2 # 不存在环路下的最小生成树 按每条边都是红色糖线来计算,输入的边为黑色糖线
    f=list(range(N))
    for i in range(M):
        x,y=[int(s) for s in input().split(" ")]
        x=x-1
        y=y-1
        u=find(x) # 寻找祖先节点
        v=find(y)
        if u==v: # 有相同的祖先节点,再连接即为环路,即该边不属于最小生成树,跳过该边,进行下一组边的判断
            continue
        f[u]=v
        ans=ans-1
    print("Case #{}: {}".format(k, ans))  


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值