Алгоритм Прима: построение 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):

Новые блоги и статьи Все статьи Все блоги /
Midnight Chicago Blues kumehtar 24.03.2026 Такой Midnight Chicago Blues, знаешь?. . Когда вечерние улицы становятся ночными, а ты не можешь уснуть. Ты идёшь в любимый старый бар, и бармен наливает тебе виски. Ты смотришь на пролетающие. . .	Контроль уникальности заводского номера - вариант №2 Maks 24.03.2026 В отличие от предыдущего варианта добавлено прерывание циклов, также добавлены новые переменные для сохранения контекста ошибки перед прерыванием цикла: Процедура ПередЗаписью(Отказ, РежимЗаписи,. . .	SDL3 для Desktop (MinGW): Вывод текста со шрифтом TTF с помощью библиотеки SDL3_ttf на Си и C++ 8Observer8 24.03.2026 Содержание блога Финальные проекты на Си и на C++: finish-text-sdl3-c. zip finish-text-sdl3-cpp. zip	Жизнь в неопределённости kumehtar 23.03.2026 Жизнь — это постоянное существование в неопределённости. Например, даже если у тебя есть список дел, невозможно дойти до точки, где всё окончательно завершено и больше ничего не осталось. В принципе,. . .
Модель здравоСохранения: работники работают быстрее после её введения. anaschu 23.03.2026 geJalZw1fLo Корпорация до введения программа здравоохранения имела много невыполненных работниками заданий, после введения программы количество заданий выросло. Но на выплатах по больничным это. . .	Контроль уникальности заводского номера - вариант №1 Maks 23.03.2026 Алгоритм контроля уникальности заводского (или серийного) номера на примере документа выдачи шин для спецтехники с табличной частью. Данные берутся из регистра сведений, по которому настроено. . .	Хочу заставить корпорации вкладываться в здоровье сотрудников: делаю мат модель здравосохранения anaschu 22.03.2026 e7EYtONaj8Y Z4Tv2zpXVVo https:/ / github. com/ shumilovas/ med2. git	Программный отбор элементов справочника по группе Maks 22.03.2026 Установка программного отбора элементов справочника "Номенклатура" из модуля формы документа. В качестве фильтра для отбора справочника служит группа номенклатуры. Отбор по наименованию группы. . .

@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