首页 > 试题广场 >

继续畅通工程

[编程题]继续畅通工程

热度指数：10018 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 64M，其他语言128M
算法知识视频讲解

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

输入描述:

    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

    当N为0时输入结束。

输出描述:

    每个测试用例的输出占一行，输出全省畅通需要的最低成本。

示例1

输入

输出

3
1
0

算法知识视频讲解

Python

TimeMac

在输入时就对已建好的路进行树的生成，且不保存，而后只保存未开始修建的路，然后进行排序和树的继续生成。

def findRoot(village):
    global villages
    if villages[village] == -1:
        return village
    else:
        temp = findRoot(villages[village])
        villages[village] = temp
        return temp
while True:
    try:
        n = int(input())
        if n == 0:
            break
        villages = [-1 for i in range(n+1)]
        nodeNum = 1
        roads = []
        for i in range(n*(n-1)//2):
            temp = list(map(int, input().split()))
            if temp[3] == 1:
                a = findRoot(temp[0])
                b = findRoot(temp[1])
                if a != b:
                    villages[a] = b
                    nodeNum += 1
            else:
                roads.append(temp)
        costs = 0
        roads.sort(key=lambda x:x[2])
        for i in range(len(roads)):
            a = findRoot(roads[i][0])
            b = findRoot(roads[i][1])
            if a != b:
                villages[a] = b
                costs += roads[i][2]
                nodeNum += 1
            if nodeNum == n:
                break
        print(costs)
    except Exception:
        break

编辑于 2018-09-28 16:01:24 回复(0)