Алгоритм Прима: построение min остовного дерева взвешенного связного неориентированного графа (Си -> Python)

@olalalalalalala · Регистрация: 14.03.2017

Студворк — интернет-сервис помощи студентам

задача: Алгоритм Прима.
Алгоритм Прима — алгоритм построения минимального остовного дерева взвешенного связного неориентированного графа.
переписала код с Си в Python.
не знаю где ошибка. как можна еще записать в питоне бесконечность?

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from __future__ import print_function
import math
n = int(input("Enter number vertices -> "))
a = {}
i = 1
j = 1
for i in range(n):
    for j in range(n):
        a[i, j] = int(input())
 
for i in range(n):
    print(*[a[i, j] for j in range(n)])
 
ne = 1
path_index = 0
path = {}
mincost = 0
visited = {}
while (ne < n):
    mi = float("inf")
    for i in range(n):
        for j in range(n):
            if (mi < a[i, j]):
                if (visited[i] != 0):
                    mi = a[i, j]
                    ka = u = i
                    b = v = j
    if (visited[u] == 0 and visited[v] == 0):
        path[path_index] = b
        path_index +=1
        ne +=1
        mincost += mi
        visited[b] = 1
 
    a[ka, b] = a[b, ka] = float('inf')
 
print(1, '-->')
k = 0
for k in range (n):
        print(path[i])
        if (i < n - 2):
            print(" --> ")
print("\n min price  ", mincost)

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#include<conio.h>
#include<iostream>
#include <locale.h>
 
using namespace std;
int a, b, u, v, n, i, j, ne = 1;
int visited[10] = { 0 }, mi, mincost = 0, cost[10][10];
 
int main()
{
    int path[100] = { 0 };
    int path_index = 0;
    setlocale(0, "RUS");
 
    cout << " Введите количество вершин "; cin >> n;
    cout << " Введите матрицу смежности \n";
 
    for (i = 1; i <= n; i++)
        for (j = 1; j <= n; j++)
        {
            cin >> cost[i][j];
            if (cost[i][j] == 0)
                cost[i][j] = 999;
        }
    visited[1] = 1;
    cout << "\n";
 
    while (ne < n)
    {
        mi = 999;
        for (int i = 1; i <= n; i++)
            for (j = 1; j <= n; j++)
                if (cost[i][j]< mi)
                    if (visited[i] != 0)
                    {
                        mi = cost[i][j];
                        a = u = i;
                        b = v = j;
                    }
        if (visited[u] == 0 || visited[v] == 0)
        {
            path[path_index] = b;
            path_index++;
            ne++;
            mincost += mi;
            visited[b] = 1;
 
        }
        cost[a][b] = cost[b][a] = 999;
    }
 
 
    cout << "\n";
 
    cout << 1 << " --> ";
    for (int i = 0; i<n - 1; i++)
    {
        cout << path[i];
        if (i<n - 2) cout << " --> ";
    }
 
    cout << "\n Минимальная стоимость  " << mincost;
 
 
    cin.get();
    cin.get();
}

@shsv · 08.05.2017, 19:58

Сравнивая с С++ кодом, сразу бросаются в глаза два несоответствия:
1. строка 23 против строки 33:

if (mi < a[i, j]): <=> if (cost[i][j]< mi) (да и без того ясно, что mi < a[i, j] всегда False);

2. строка 28 против строки 40:

if (visited[u] == 0 and visited[v] == 0) <=> if (visited[u] == 0 || visited[v] == 0) (and vs or)

@olalalalalalala · 08.05.2017, 22:38 **[ТС]**

а как эту ошибку исправить?

@olalalalalalala · 08.05.2017, 22:46 **[ТС]**

а как эту ошибку исправить?

oldnewyear · 09.05.2017, 02:25

Лучше сделать visited множеством

Python
1
visited = set()

И проверять, есть ли в нем значение так:

Python
1
if (i in visited):

Добавлено через 3 минуты
Добавлять в set можно так:

Python
1
visited.add(i)

Добавлено через 36 минут
в 39 строке должно быть

Python
1
for i in range (n):

Новые блоги и статьи Все статьи Все блоги /
сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/	WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .	Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .	Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .
Модель микоризы: классовый агентный подход 3 anaschu 06.01.2026 aa0a7f55b50dd51c5ec569d2d10c54f6/ O1rJuneU_ls https:/ / vkvideo. ru/ video-115721503_456239114	Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ФедосеевПавел 06.01.2026 Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ВВЕДЕНИЕ Введу сокращения: аналоговый ПИД — ПИД регулятор с управляющим выходом в виде числа в диапазоне от 0% до. . .	Модель микоризы: классовый агентный подход 2 anaschu 06.01.2026 репозиторий https:/ / github. com/ shumilovas/ fungi ветка по-частям. коммит Create переделка под биомассу. txt вход sc, но sm считается внутри мицелия. кстати, обьем тоже должен там считаться. . . .	Расчёт токов в цепи постоянного тока igorrr37 05.01.2026 / * Дана цепь постоянного тока с сопротивлениями и источниками (напряжения, ЭДС и тока). Найти токи и напряжения во всех элементах. Программа составляет систему уравнений по 1 и 2 законам Кирхгофа и. . .

@shsv 757 / 306 / 190 Регистрация: 20.05.2016 Сообщений: 593
	08.05.2017, 19:58
	Сравнивая с С++ кодом, сразу бросаются в глаза два несоответствия: 1. строка 23 против строки 33: `if (mi < a[i, j]):` <=> `if (cost[i][j]< mi)` (да и без того ясно, что mi < a[i, j] всегда False); 2. строка 28 против строки 40: `if (visited[u] == 0 and visited[v] == 0)` <=> `if (visited[u] == 0 \|\| visited[v] == 0)` (and vs or) 1

@olalalalalalala 0 / 0 / 0 Регистрация: 14.03.2017 Сообщений: 21
	08.05.2017, 22:38 [ТС]
	а как эту ошибку исправить? 0

@olalalalalalala 0 / 0 / 0 Регистрация: 14.03.2017 Сообщений: 21
	08.05.2017, 22:46 [ТС]
	а как эту ошибку исправить? Миниатюры 0