Algoritmo de Prim

Neste tutorial, você aprenderá como funciona o algoritmo de Prim. Além disso, você encontrará exemplos de trabalho do algoritmo de Prim em C, C ++, Java e Python.

O algoritmo de Prim é um algoritmo de spanning tree mínimo que pega um gráfico como entrada e encontra o subconjunto das arestas desse gráfico que

forma uma árvore que inclui todos os vértices
tem a soma mínima de pesos entre todas as árvores que podem ser formadas a partir do gráfico

Como funciona o algoritmo de Prim

Ele se enquadra em uma classe de algoritmos chamados algoritmos gananciosos que encontram o ótimo local na esperança de encontrar um ótimo global.

Começamos a partir de um vértice e continuamos adicionando arestas com o menor peso até atingir nosso objetivo.

As etapas para implementar o algoritmo de Prim são as seguintes:

Inicialize a árvore geradora mínima com um vértice escolhido aleatoriamente.
Encontre todas as arestas que conectam a árvore aos novos vértices, encontre o mínimo e adicione-o à árvore
Continue repetindo a etapa 2 até obtermos uma árvore geradora mínima

Exemplo de algoritmo de Prim

Comece com um gráfico ponderado

Escolha um vértice

Escolha a aresta mais curta deste vértice e adicione-a

Escolha o vértice mais próximo que ainda não está na solução

Escolha a aresta mais próxima que ainda não está na solução, se houver várias escolhas, escolha uma ao acaso

Repita até você tem uma árvore geradora

Pseudocódigo do algoritmo de Prim

O pseudocódigo para o algoritmo de prim mostra como criamos dois conjuntos de vértices U e VU. U contém a lista de vértices que foram visitados e VU a lista de vértices que não foram visitados. Um por um, movemos os vértices do conjunto VU para o conjunto U conectando a aresta de menor peso.

 T = ∅; U = ( 1 ); while (U ≠ V) let (u, v) be the lowest cost edge such that u ∈ U and v ∈ V - U; T = T ∪ ((u, v)) U = U ∪ (v)

Exemplos de Python, Java e C / C ++

Embora a representação da matriz de adjacência de grafos seja usada, este algoritmo também pode ser implementado usando Adjacency List para melhorar sua eficiência.

Python Java C C ++

 # Prim's Algorithm in Python INF = 9999999 # number of vertices in graph V = 5 # create a 2d array of size 5x5 # for adjacency matrix to represent graph G = ((0, 9, 75, 0, 0), (9, 0, 95, 19, 42), (75, 95, 0, 51, 66), (0, 19, 51, 0, 31), (0, 42, 66, 31, 0)) # create a array to track selected vertex # selected will become true otherwise false selected = (0, 0, 0, 0, 0) # set number of edge to 0 no_edge = 0 # the number of egde in minimum spanning tree will be # always less than(V - 1), where V is number of vertices in # graph # choose 0th vertex and make it true selected(0) = True # print for edge and weight print("Edge : Weight") while (no_edge  G(i)(j): minimum = G(i)(j) x = i y = j print(str(x) + "-" + str(y) + ":" + str(G(x)(y))) selected(y) = True no_edge += 1

 // Prim's Algorithm in Java import java.util.Arrays; class PGraph ( public void Prim(int G()(), int V) ( int INF = 9999999; int no_edge; // number of edge // create a array to track selected vertex // selected will become true otherwise false boolean() selected = new boolean(V); // set selected false initially Arrays.fill(selected, false); // set number of edge to 0 no_edge = 0; // the number of egde in minimum spanning tree will be // always less than (V -1), where V is number of vertices in // graph // choose 0th vertex and make it true selected(0) = true; // print for edge and weight System.out.println("Edge : Weight"); while (no_edge < V - 1) ( // For every vertex in the set S, find the all adjacent vertices // , calculate the distance from the vertex selected at step 1. // if the vertex is already in the set S, discard it otherwise // choose another vertex nearest to selected vertex at step 1. int min = INF; int x = 0; // row number int y = 0; // col number for (int i = 0; i < V; i++) ( if (selected(i) == true) ( for (int j = 0; j G(i)(j)) ( min = G(i)(j); x = i; y = j; ) ) ) ) ) System.out.println(x + " - " + y + " : " + G(x)(y)); selected(y) = true; no_edge++; ) ) public static void main(String() args) ( PGraph g = new PGraph(); // number of vertices in grapj int V = 5; // create a 2d array of size 5x5 // for adjacency matrix to represent graph int()() G = ( ( 0, 9, 75, 0, 0 ), ( 9, 0, 95, 19, 42 ), ( 75, 95, 0, 51, 66 ), ( 0, 19, 51, 0, 31 ), ( 0, 42, 66, 31, 0 ) ); g.Prim(G, V); ) )

 // Prim's Algorithm in C #include #include #define INF 9999999 // number of vertices in graph #define V 5 // create a 2d array of size 5x5 //for adjacency matrix to represent graph int G(V)(V) = ( (0, 9, 75, 0, 0), (9, 0, 95, 19, 42), (75, 95, 0, 51, 66), (0, 19, 51, 0, 31), (0, 42, 66, 31, 0)); int main() ( int no_edge; // number of edge // create a array to track selected vertex // selected will become true otherwise false int selected(V); // set selected false initially memset(selected, false, sizeof(selected)); // set number of edge to 0 no_edge = 0; // the number of egde in minimum spanning tree will be // always less than (V -1), where V is number of vertices in //graph // choose 0th vertex and make it true selected(0) = true; int x; // row number int y; // col number // print for edge and weight printf("Edge : Weight"); while (no_edge < V - 1) ( //For every vertex in the set S, find the all adjacent vertices // , calculate the distance from the vertex selected at step 1. // if the vertex is already in the set S, discard it otherwise //choose another vertex nearest to selected vertex at step 1. int min = INF; x = 0; y = 0; for (int i = 0; i < V; i++) ( if (selected(i)) ( for (int j = 0; j G(i)(j)) ( min = G(i)(j); x = i; y = j; ) ) ) ) ) printf("%d - %d : %d", x, y, G(x)(y)); selected(y) = true; no_edge++; ) return 0; )

 // Prim's Algorithm in C++ #include #include using namespace std; #define INF 9999999 // number of vertices in grapj #define V 5 // create a 2d array of size 5x5 //for adjacency matrix to represent graph int G(V)(V) = ( (0, 9, 75, 0, 0), (9, 0, 95, 19, 42), (75, 95, 0, 51, 66), (0, 19, 51, 0, 31), (0, 42, 66, 31, 0)); int main() ( int no_edge; // number of edge // create a array to track selected vertex // selected will become true otherwise false int selected(V); // set selected false initially memset(selected, false, sizeof(selected)); // set number of edge to 0 no_edge = 0; // the number of egde in minimum spanning tree will be // always less than (V -1), where V is number of vertices in //graph // choose 0th vertex and make it true selected(0) = true; int x; // row number int y; // col number // print for edge and weight cout << "Edge" << " : " << "Weight"; cout << endl; while (no_edge < V - 1) ( //For every vertex in the set S, find the all adjacent vertices // , calculate the distance from the vertex selected at step 1. // if the vertex is already in the set S, discard it otherwise //choose another vertex nearest to selected vertex at step 1. int min = INF; x = 0; y = 0; for (int i = 0; i < V; i++) ( if (selected(i)) ( for (int j = 0; j G(i)(j)) ( min = G(i)(j); x = i; y = j; ) ) ) ) ) cout << x << " - " << y << " : " << G(x)(y); cout << endl; selected(y) = true; no_edge++; ) return 0; )

Algoritmo de Prim vs Kruskal

O algoritmo de Kruskal é outro algoritmo de árvore geradora mínima popular que usa uma lógica diferente para encontrar o MST de um gráfico. Em vez de começar a partir de um vértice, o algoritmo de Kruskal classifica todas as arestas de baixo peso a alto e continua adicionando as arestas mais baixas, ignorando aquelas arestas que criam um ciclo.