import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.*;
import java.util.*;

class LRMCmkpeeling {

    // experiment constants from the paper
    // n in [20k, 1M], 10 planted clusters, p_intra in [0.005, 0.02], q in [1e-4, 1e-3]
    static final int NUM_CLUSTERS = 10;
    static final double CLUSTER_FRACTION = 0.20; // 20% of nodes participate in planted clusters

    // Use three intra densities across the paper's range, with a fixed sparse q
    // Main runtime figure can use the middle setting: p_intra = 0.01, q = 1e-4
    static final double[] PINTRA_SERIES = {0.010};
    static final double PINTER_FIXED = 1e-4; // keep inter-links sparse and tractable at n up to 1M
    static final String[] SERIES_LABELS = {"p0.010"};

    // Trials and sizes
    static final int S_MIN = 10_000;
    static final int S_MAX = 1_050_000;
    static final int NUM_SIZES = 30;
    static final int TRIALS = 3;

    // Algorithm hyperparameters
    static final double EPSILON = 1e-6; // forwarded to LRMC main if PASS_EPSILON is true

    // JVM settings for the inner run; bump if you hit OOM at the top end
    static final String EXTRA_HEAP = "-Xmx8g";

    // If your LRMC main expects epsilon before the input path, set this true
    static final boolean PASS_EPSILON = true;

    static final long SEED = 123456789L;

    static String lrmcMain;
    static String outputCsvFile;

    public static void main(String[] args) throws Exception {
        if (args.length < 2) {
            System.err.println("Usage: java LRMCmkpaper <LRMC_MAIN> <output_csv_file>");
            return;
        }

        lrmcMain = args[0];
        outputCsvFile = args[1];

        Random rng = new Random(SEED);
        List<Row> allRows = new ArrayList<>();
        int[] sizes = logSpaced(S_MIN, S_MAX, NUM_SIZES);

        for (int si = 0; si < PINTRA_SERIES.length; si++) {
            String series = SERIES_LABELS[si];
            double pIntra = PINTRA_SERIES[si];
            double pInter = PINTER_FIXED;

            for (int n : sizes) {
                Path inputFile = Files.createTempFile("clique2_input_" + series + "_n" + n + "_", ".txt");
                inputFile.toFile().deleteOnExit();

                long m = generateClusteredGraphToFile(
                        n, NUM_CLUSTERS, CLUSTER_FRACTION,
                        pIntra, pInter, rng, inputFile);

                for (int t = 0; t < TRIALS; t++) {
                    double msLRMC = runLRMC(EPSILON, inputFile);
                    double msCharikar = runCharikarPeeling(inputFile);
                    allRows.add(new Row(series, n, m, t + 1, msLRMC, msCharikar, pIntra, pInter));
                }
            }
        }

        // Write CSV to specified file
        try (BufferedWriter writer = Files.newBufferedWriter(Paths.get(outputCsvFile), StandardCharsets.UTF_8)) {
            writer.write("series,n,m,trial,ms_lrmc,ms_charikar,ratio_char_over_lrmc,p_intra,p_inter\n");
            for (Row r : allRows) {
                double ratio = r.msCharikar / Math.max(1e-9, r.msLRMC);
                writer.write(String.format(Locale.US, "%s,%d,%d,%d,%.3f,%.3f,%.4f,%.6f,%.6g\n",
                        r.series, r.n, r.m, r.trial, r.msLRMC, r.msCharikar, ratio, r.pIntra, r.pInter));
            }
        }

        // Optional: also echo summaries to stdout
        Map<String, Map<Integer, List<Row>>> bySeriesSize = new TreeMap<>();
        for (Row r : allRows) {
            bySeriesSize.computeIfAbsent(r.series, s -> new TreeMap<>())
                    .computeIfAbsent(r.n, _k -> new ArrayList<>()).add(r);
        }
        for (var eSeries : bySeriesSize.entrySet()) {
            String s = eSeries.getKey();
            for (var e : eSeries.getValue().entrySet()) {
                int n = e.getKey();
                long m = e.getValue().get(0).m;

                double[] arrLR = e.getValue().stream().mapToDouble(rr -> rr.msLRMC).toArray();
                double[] arrCH = e.getValue().stream().mapToDouble(rr -> rr.msCharikar).toArray();
                double meanLR = mean(arrLR), sdLR = stddev(arrLR, meanLR);
                double meanCH = mean(arrCH), sdCH = stddev(arrCH, meanCH);
                double ratio = meanCH / Math.max(1e-9, meanLR);

                double pIntra = e.getValue().get(0).pIntra;
                double pInter = e.getValue().get(0).pInter;

                System.out.printf(Locale.US, "# summary,%s,%d,%d,mean_lrmc_ms=%.3f,mean_char_ms=%.3f,ratio=%.4f,p_intra=%.6f,p_inter=%.6g%n",
                        s, n, m, meanLR, meanCH, ratio, pIntra, pInter);
                if (TRIALS > 1) {
                    System.out.printf(Locale.US, "# summary_std,%s,%d,%d,sd_lrmc_ms=%.3f,sd_char_ms=%.3f%n",
                            s, n, m, sdLR, sdCH);
                }
            }
        }
    }

    // run LRMC external main
    private static double runLRMC(double epsilon, Path inputFile) throws IOException, InterruptedException {
        String javaBin = System.getProperty("java.home") + File.separator + "bin" + File.separator + "java";
        String classpath = System.getProperty("java.class.path");

        List<String> cmd = new ArrayList<>();
        cmd.add(javaBin);
        cmd.add(EXTRA_HEAP);
        cmd.add("-cp");
        cmd.add(classpath);
        cmd.add(lrmcMain);
        if (PASS_EPSILON) cmd.add(Double.toString(epsilon));
        cmd.add(inputFile.toAbsolutePath().toString());

        ProcessBuilder pb = new ProcessBuilder(cmd);
        pb.redirectErrorStream(true);
        Process p = pb.start();

        String lastRuntime = null;
        try (BufferedReader br = new BufferedReader(
                new InputStreamReader(p.getInputStream(), StandardCharsets.UTF_8))) {
            String line;
            while ((line = br.readLine()) != null) {
                if (line.startsWith("Runtime:")) lastRuntime = line;
            }
        }
        int exit = p.waitFor();
        if (exit != 0) throw new RuntimeException("LRMC main exited with code " + exit);
        if (lastRuntime == null) throw new RuntimeException("No 'Runtime: ... ms' line from LRMC main");

        String msStr = lastRuntime.replace("Runtime:", "").replace("ms", "").trim();
        System.out.println(inputFile.toAbsolutePath().toString());
        return Double.parseDouble(msStr);
    }

    // Charikar's greedy peeling (densest subgraph), end-to-end runtime
    // Includes file parsing and adjacency build for parity with LRMC's end-to-end measurement.
    private static double runCharikarPeeling(Path edgeListFile) throws IOException {
        long t0 = System.nanoTime();

        // First pass: read header and degrees
        int n;
        long mFromHeader;
        int[] deg;
        try (BufferedReader br = Files.newBufferedReader(edgeListFile, StandardCharsets.UTF_8)) {
            String hdr = br.readLine();
            if (hdr == null) throw new IOException("Empty graph file");
            int sp = hdr.indexOf(' ');
            if (sp <= 0) throw new IOException("Bad header: " + hdr);
            n = Integer.parseInt(hdr.substring(0, sp));
            mFromHeader = Long.parseLong(hdr.substring(sp + 1));
            deg = new int[n];

            String s;
            while ((s = br.readLine()) != null) {
                if (s.isEmpty()) continue;
                int sp2 = s.indexOf(' ');
                if (sp2 <= 0) continue;
                int u = Integer.parseInt(s.substring(0, sp2)) - 1;
                int v = Integer.parseInt(s.substring(sp2 + 1)) - 1;
                if (u == v || u < 0 || u >= n || v < 0 || v >= n) continue;
                deg[u]++; deg[v]++;
            }
        }

        int maxDeg = 0;
        long totalAdj = 0;
        for (int d : deg) { if (d > maxDeg) maxDeg = d; totalAdj += d; }

        // Build adjacency with offsets, second pass to fill
        int[] off = new int[n + 1];
        for (int i = 0; i < n; i++) off[i + 1] = off[i] + deg[i];
        int[] adj = new int[(int) totalAdj];
        int[] cur = Arrays.copyOf(off, off.length);

        try (BufferedReader br = Files.newBufferedReader(edgeListFile, StandardCharsets.UTF_8)) {
            br.readLine(); // skip header
            String s;
            while ((s = br.readLine()) != null) {
                if (s.isEmpty()) continue;
                int sp = s.indexOf(' ');
                if (sp <= 0) continue;
                int u = Integer.parseInt(s.substring(0, sp)) - 1;
                int v = Integer.parseInt(s.substring(sp + 1)) - 1;
                if (u == v || u < 0 || u >= n || v < 0 || v >= n) continue;
                adj[cur[u]++] = v;
                adj[cur[v]++] = u;
            }
        }

        // Greedy peeling with integer buckets
        int[] curDeg = Arrays.copyOf(deg, deg.length);
        boolean[] alive = new boolean[n];
        Arrays.fill(alive, true);
        int[] head = new int[maxDeg + 1];
        int[] tail = new int[maxDeg + 1];
        Arrays.fill(head, -1);
        Arrays.fill(tail, -1);
        int[] next = new int[n];
        int[] prev = new int[n];
        Arrays.fill(next, -1);
        Arrays.fill(prev, -1);

        for (int v = 0; v < n; v++) bucketAdd(v, curDeg[v], head, tail, next, prev);

        int currentMin = 0;
        long aliveEdges = totalAdj / 2; // equals m
        int aliveCount = n;
        double bestDensity = aliveCount > 0 ? aliveEdges / (double) aliveCount : 0.0;

        for (int removed = 0; removed < n; removed++) {
            while (currentMin <= maxDeg && head[currentMin] == -1) currentMin++;
            if (currentMin > maxDeg) break; // no vertices left

            int v = head[currentMin];
            bucketRemove(v, currentMin, head, tail, next, prev);
            int dv = curDeg[v];
            alive[v] = false;

            // update neighbors
            for (int p = off[v]; p < off[v + 1]; p++) {
                int u = adj[p];
                if (!alive[u]) continue;
                int du = curDeg[u];
                if (du <= 0) continue;
                bucketRemove(u, du, head, tail, next, prev);
                curDeg[u] = du - 1;
                bucketAdd(u, du - 1, head, tail, next, prev);
            }

            aliveEdges -= dv;
            aliveCount--;
            if (aliveCount > 0) {
                double dens = aliveEdges / (double) aliveCount;
                if (dens > bestDensity) bestDensity = dens;
            }

            // neighbors can drop the min by at most 1
            currentMin = Math.max(0, currentMin - 1);
        }

        long t1 = System.nanoTime();
        return (t1 - t0) / 1e6; // ms
    }

    private static void bucketAdd(int v, int d, int[] head, int[] tail, int[] next, int[] prev) {
        // insert at head of bucket d
        int h = head[d];
        prev[v] = -1;
        next[v] = h;
        if (h != -1) prev[h] = v; else tail[d] = v;
        head[d] = v;
    }

    private static void bucketRemove(int v, int d, int[] head, int[] tail, int[] next, int[] prev) {
        int pv = prev[v];
        int nv = next[v];
        if (pv != -1) next[pv] = nv; else head[d] = nv;
        if (nv != -1) prev[nv] = pv; else tail[d] = pv;
        prev[v] = -1; next[v] = -1;
    }

    // clustered generator (expected O(m))
    static long generateClusteredGraphToFile(
            int n, int k, double frac, double pIntra, double pInter, Random rng, Path outFile) throws IOException {

        int clusterTotal = (int) Math.round(frac * n);
        int[] nodes = new int[n];
        for (int i = 0; i < n; i++) nodes[i] = i + 1;
        shuffle(nodes, rng);

        int base = clusterTotal / k, rem = clusterTotal % k;
        int[][] clusters = new int[k][];
        int idx = 0;
        for (int i = 0; i < k; i++) {
            int sz = base + (i < rem ? 1 : 0);
            clusters[i] = Arrays.copyOfRange(nodes, idx, idx + sz);
            Arrays.sort(clusters[i]);
            idx += sz;
        }
        int[] background = Arrays.copyOfRange(nodes, idx, n);
        Arrays.sort(background);

        Path tmpEdges = Files.createTempFile("edges_only_", ".txt");
        tmpEdges.toFile().deleteOnExit();
        long m = 0;

        try (BufferedWriter w = Files.newBufferedWriter(tmpEdges, StandardCharsets.UTF_8)) {
            // intra-cluster edges
            for (int i = 0; i < k; i++) m += triPairsToWriter(clusters[i], pIntra, w, rng);
            // inter-links between clusters
            for (int i = 0; i < k; i++) for (int j = i + 1; j < k; j++)
                m += rectPairsToWriter(clusters[i], clusters[j], pInter, w, rng);
            // links between clusters and background
            for (int i = 0; i < k; i++) m += rectPairsToWriter(clusters[i], background, pInter, w, rng);
            // sparse background background links as well
            m += triPairsToWriter(background, pInter, w, rng);
        }

        try (BufferedWriter hdr = Files.newBufferedWriter(outFile, StandardCharsets.UTF_8,
                StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING)) {
            hdr.write(n + " " + m);
            hdr.newLine();
        }
        try (OutputStream out = Files.newOutputStream(outFile, StandardOpenOption.APPEND);
             InputStream in = Files.newInputStream(tmpEdges)) {
            byte[] buf = new byte[1 << 20];
            int len;
            while ((len = in.read(buf)) != -1) out.write(buf, 0, len);
        }
        return m;
    }

    // skip-sampling over unordered pairs in a set
    static long triPairsToWriter(int[] set, double p, BufferedWriter w, Random rng) throws IOException {
        int s = set.length; if (s < 2 || p <= 0) return 0L;
        final double logq = Math.log(1.0 - p);
        long written = 0;
        int row = 0, off = -1;
        while (row < s - 1) {
            double r = rng.nextDouble();
            int skip = (int) Math.floor(Math.log(1.0 - r) / logq);
            off += 1 + skip;
            while (row < s - 1 && off >= (s - row - 1)) {
                off -= (s - row - 1);
                row++;
            }
            if (row < s - 1) {
                int u = set[row], v = set[row + 1 + off];
                w.write(u + " " + v); w.newLine();
                written++;
            }
        }
        return written;
    }

    // skip-sampling over A x B
    static long rectPairsToWriter(int[] A, int[] B, double p, BufferedWriter w, Random rng) throws IOException {
        int a = A.length, b = B.length; if (a == 0 || b == 0 || p <= 0) return 0L;
        final double logq = Math.log(1.0 - p);
        long total = 1L * a * b, t = -1, written = 0;
        while (true) {
            double r = rng.nextDouble();
            long skip = (long) Math.floor(Math.log(1.0 - r) / logq);
            t += 1 + skip;
            if (t >= total) break;
            int i = (int) (t / b), j = (int) (t % b);
            w.write(A[i] + " " + B[j]); w.newLine();
            written++;
        }
        return written;
    }

    // helpers
    static int[] logSpaced(int lo, int hi, int k) {
        double a = Math.log(lo), b = Math.log(hi);
        int[] out = new int[k];
        for (int i = 0; i < k; i++) {
            double t = i / (double) (k - 1);
            out[i] = (int) Math.round(Math.exp(a + t * (b - a)));
            out[i] = Math.max(lo, Math.min(hi, (out[i] + 500) / 1000 * 1000)); // snap to nearest 1000
        }
        for (int i = 1; i < k; i++) if (out[i] <= out[i - 1]) out[i] = out[i - 1] + 1000;
        out[k - 1] = hi;
        return out;
    }

    static void shuffle(int[] a, Random rng) {
        for (int i = a.length - 1; i > 0; i--) { int j = rng.nextInt(i + 1); int t = a[i]; a[i] = a[j]; a[j] = t; }
    }

    static double mean(double[] x) { double s = 0; for (double v : x) s += v; return s / x.length; }
    static double stddev(double[] x, double mean) {
        if (x.length <= 1) return 0;
        double s2 = 0; for (double v : x) { double d = v - mean; s2 += d * d; }
        return Math.sqrt(s2 / (x.length - 1));
    }

    static final class Row {
        final String series;
        final int n;
        final long m;
        final int trial;
        final double msLRMC;
        final double msCharikar;
        final double pIntra;
        final double pInter;
        Row(String series, int n, long m, int trial, double msLRMC, double msCharikar, double pIntra, double pInter) {
            this.series = series; this.n = n; this.m = m; this.trial = trial;
            this.msLRMC = msLRMC; this.msCharikar = msCharikar;
            this.pIntra = pIntra; this.pInter = pInter;
        }
    }
}