forked from TheAlgorithms/Java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathStoerWagner.java
More file actions
78 lines (65 loc) · 2.47 KB
/
StoerWagner.java
File metadata and controls
78 lines (65 loc) · 2.47 KB
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
68
69
70
71
72
73
74
75
76
77
78
package com.thealgorithms.graph;
/**
* An implementation of the Stoer-Wagner algorithm to find the global minimum cut of an undirected, weighted graph.
* A minimum cut is a partition of the graph's vertices into two disjoint sets with the minimum possible edge weight
* sum connecting the two sets.
*
* Wikipedia: https://en.wikipedia.org/wiki/Stoer%E2%80%93Wagner_algorithm
* Time Complexity: O(V^3) where V is the number of vertices.
*/
public class StoerWagner {
/**
* Finds the minimum cut in the given undirected, weighted graph.
*
* @param graph An adjacency matrix representing the graph. graph[i][j] is the weight of the edge between i and j.
* @return The weight of the minimum cut.
*/
public int findMinCut(int[][] graph) {
int n = graph.length;
if (n < 2) {
return 0;
}
int[][] currentGraph = new int[n][n];
for (int i = 0; i < n; i++) {
System.arraycopy(graph[i], 0, currentGraph[i], 0, n);
}
int minCut = Integer.MAX_VALUE;
boolean[] merged = new boolean[n];
for (int phase = 0; phase < n - 1; phase++) {
boolean[] inSetA = new boolean[n];
int[] weights = new int[n];
int prev = -1;
int last = -1;
for (int i = 0; i < n - phase; i++) {
int maxWeight = -1;
int currentVertex = -1;
for (int j = 0; j < n; j++) {
if (!merged[j] && !inSetA[j] && weights[j] > maxWeight) {
maxWeight = weights[j];
currentVertex = j;
}
}
if (currentVertex == -1) {
// This can happen if the graph is disconnected.
return 0;
}
prev = last;
last = currentVertex;
inSetA[last] = true;
for (int j = 0; j < n; j++) {
if (!merged[j] && !inSetA[j]) {
weights[j] += currentGraph[last][j];
}
}
}
minCut = Math.min(minCut, weights[last]);
// Merge 'last' vertex into 'prev' vertex
for (int i = 0; i < n; i++) {
currentGraph[prev][i] += currentGraph[last][i];
currentGraph[i][prev] = currentGraph[prev][i];
}
merged[last] = true;
}
return minCut;
}
}