File size: 6,298 Bytes
bf620c6 |
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 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 |
import java.io.FileReader;
import java.io.IOException;
import java.util.*;
public class clique2 {
public static int n;
public static void main(String[] args) throws IOException {
Scanner r = new Scanner(new FileReader(args[1]));
double epsilon = Double.parseDouble(args[0]);
n = r.nextInt();
int m = r.nextInt();
long start = System.nanoTime();
TreeMap<Integer, ArrayList<Integer>> map = new TreeMap<>();
PriorityQueue<Pair> pq = new PriorityQueue<>();
int[] degrees = new int[n+1];
Stack<Pair> stack = new Stack<>();
for (int i = 0; i < n; i++) {
map.put(i + 1, new ArrayList<>());
}
for (int i = 0; i < m; i++) { // Read in connections
int node1 = r.nextInt();
int node2 = r.nextInt();
map.get(node1).add(node2);
map.get(node2).add(node1);
}
for (int i = 1; i < n + 1; i++) {
pq.add(new Pair(i, map.get(i).size()));
degrees[i] = map.get(i).size();
}
// Remove nodes one by one
while(!pq.isEmpty()) {
Pair minDegreeNode = pq.poll();
// Ignore outdated nodes
if (degrees[minDegreeNode.node] != minDegreeNode.degree) continue;
for (int connectedNode : map.get(minDegreeNode.node)) {
if (degrees[connectedNode] > 0) {
// Remove the current minDegreeNode and update neighbor's nodes
degrees[connectedNode] -= 1;
pq.add(new Pair(connectedNode, degrees[connectedNode]));
}
}
degrees[minDegreeNode.node] = 0;
stack.add(minDegreeNode);
}
// Add back nodes
DSU dsu = new DSU();
double sMax = 0;
int uStar = 0;
long[] currLaplacian = new long[n+1]; // Stores Laplacian values for each component
boolean[] readded = new boolean[n+1]; // Tracks which nodes have been added back
// Process nodes in reverse order from stack
while (!stack.isEmpty()) {
Pair u = stack.pop();
// Find all neighbors that have already been processed (added back to graph)
ArrayList<Integer> uN = new ArrayList<>();
for (int v : map.get(u.node)) {
if (readded[v]) {
uN.add(v);
}
}
// Collect all unique component roots from processed neighbors
HashSet<Integer> roots = new HashSet<>();
for (int v : uN) {
int rv = dsu.find(v);
roots.add(rv);
// Update Laplacian based on degree differences within the component
int dv = degrees[v];
for (int x : map.get(v)) {
if (readded[x]) {
int dx = degrees[x];
currLaplacian[rv] += 2L * (dv - dx) + 1L;
}
}
degrees[v] += 1;
}
// Update degree of current node to number of processed neighbors so far
int du = uN.size();
degrees[u.node] = du;
// Calculate Laplacian contribution from new edges (based on degree differences)
long newEdge = 0;
for (int v : uN) {
int dv = degrees[v];
long degreeDiff = (long) (du - dv);
newEdge += degreeDiff * degreeDiff;
}
// Sum up Laplacian values from all neighbor components
long vLaplacian = 0;
for (int rv : roots) {
vLaplacian += currLaplacian[rv];
}
// Merge current node with all its processed neighbors into one component
int s = 0;
for (int v : uN) {
s = dsu.union(u.node, v);
}
// Update Laplacian for the merged component
int mergedComponentRoot = dsu.find(u.node);
currLaplacian[mergedComponentRoot] += vLaplacian + newEdge;
readded[u.node] = true;
// Calculate density score
double sL = s / (currLaplacian[mergedComponentRoot] + epsilon);
if (sL > sMax) {
sMax = sL;
uStar = mergedComponentRoot;
}
}
System.out.println(sMax + ", " + uStar);
long end = System.nanoTime();
double elapsedMs = (end - start) / 1_000_000.0;
System.out.printf("Runtime: %.3f ms%n", elapsedMs);
}
static class DSU {
public int[] parents = new int[n+1];
public int[] size = new int[n+1];
public DSU() {
for (int i = 1; i < n+1; i++) {
size[i] = 1;
parents[i] = i;
}
}
public int find(int x) {
if (parents[x] !=x) {
parents[x] = find(parents[x]);
}
return parents[x];
}
public int union(int n1, int n2) {
int pn1 = find(n1);
int pn2 = find(n2);
if (pn1 == pn2) return 0;
if (size[pn1] < size[pn2]) {
int temp = pn1;
pn1 = pn2;
pn2 = temp;
}
parents[pn2] = pn1;
size[pn1] += size[pn2];
return size[pn1];
}
}
static class Pair implements Comparable<Pair> {
public int node;
public int degree;
public Pair(int node, int degree) {
this.node = node;
this.degree = degree;
}
public int compareTo(Pair o) {
if (this.degree == o.degree) {
return Integer.compare(this.node, o.node);
}
return Integer.compare(this.degree, o.degree);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Pair pair = (Pair) o;
return node == pair.node && degree == pair.degree;
}
@Override
public int hashCode() {
return Objects.hash(node, degree);
}
}
}
|