/*************************************************************************
 *  Kompilacja:  javac EdgeWeightedDirectedCycle.java
 *  Wykonanie:    java EdgeWeightedDirectedCycle V E F
 *  Zależności: EdgeWeightedDigraph.java DirectedEdge Stack.java
 *
 *  Znajduje cykl skierowany w digrafie ważonym
 *  Złożoność O(E + V).
 *
 *
 *************************************************************************/

public class EdgeWeightedDirectedCycle {
    private boolean[] marked;             // marked[v] = czy wierzchołek v jest oznaczony?
    private DirectedEdge[] edgeTo;        // edgeTo[v] = poprzednia krawędź na ścieżce do v
    private boolean[] onStack;            // onStack[v] = czy wierzchołek znajduje się na stosie?
    private Stack<DirectedEdge> cycle;    // Cykl skierowany (lub null, jeśli taki cykl nie istnieje)

    public EdgeWeightedDirectedCycle(EdgeWeightedDigraph G) {
        marked  = new boolean[G.V()];
        onStack = new boolean[G.V()];
        edgeTo  = new DirectedEdge[G.V()];
        for (int v = 0; v < G.V(); v++)
            if (!marked[v]) dfs(G, v);

        // Sprawdzanie, czy digraf obejmuje cykl
        assert check(G);
    }


    // Sprawdzanie, czy algorytm wyznacza porządek topologiczny lub znajduje cykl skierowany
    private void dfs(EdgeWeightedDigraph G, int v) {
        onStack[v] = true;
        marked[v] = true;
        for (DirectedEdge e : G.adj(v)) {
            int w = e.to();

            // Kończenie pracy, jeśli znaleziono cykl skierowany
            if (cycle != null) return;

            // Znaleziono wierzchołek, dlatego należy rekurencyjnie wywołać metodę
            else if (!marked[w]) {
                edgeTo[w] = e;
                dfs(G, w);
            }

            // Odtwarzanie cyklu skierowanego
            else if (onStack[w]) {
                cycle = new Stack<DirectedEdge>();
                while (e.from() != w) {
                    cycle.push(e);
                    e = edgeTo[e.from()];
                }
                cycle.push(e);
            }
        }

        onStack[v] = false;
    }

    public boolean hasCycle()             { return cycle != null;   }
    public Iterable<DirectedEdge> cycle() { return cycle;           }


    // Sprawdzanie, czy digraf jest acykliczny lub obejmuje cykl skierowany
    private boolean check(EdgeWeightedDigraph G) {

        // Digraf ważony jest cykliczny
        if (hasCycle()) {
            // Sprawdzanie cyklu
            DirectedEdge first = null, last = null;
            for (DirectedEdge e : cycle()) {
                if (first == null) first = e;
                if (last != null) {
                    if (last.to() != e.from()) {
                        System.err.printf("Krawędzie cyklu %s i %s nie są incydentne\n", last, e);
                        return false;
                    }
                }
                last = e;
            }

            if (last.to() != first.from()) {
                System.err.printf("Krawędzie cyklu %s i %s nie są incydentne\n", last, first);
                return false;
            }
        }


        return true;
    }

    public static void main(String[] args) {

        // Tworzenie losowego grafu DAG o V wierzchołkach i E krawędziach oraz dodawanie F losowych krawędzi
        int V = Integer.parseInt(args[0]);
        int E = Integer.parseInt(args[1]);
        int F = Integer.parseInt(args[2]);
        EdgeWeightedDigraph G = new EdgeWeightedDigraph(V);
        int[] vertices = new int[V];
        for (int i = 0; i < V; i++) vertices[i] = i;
        StdRandom.shuffle(vertices);
        for (int i = 0; i < E; i++) {
            int v, w;
            do {
                v = StdRandom.uniform(V);
                w = StdRandom.uniform(V);
            } while (v >= w);
            double weight = Math.random();
            G.addEdge(new DirectedEdge(v, w, weight));
        }

        // Dodawanie F dodatkowych krawędzi
        for (int i = 0; i < F; i++) {
            int v = (int) (Math.random() * V);
            int w = (int) (Math.random() * V);
            double weight = Math.random();
            G.addEdge(new DirectedEdge(v, w, weight));
        }

        StdOut.println(G);

        // Znajdowanie cyklu skierowanego
        EdgeWeightedDirectedCycle finder = new EdgeWeightedDirectedCycle(G);
        if (finder.hasCycle()) {
            StdOut.print("Cykl: ");
            for (DirectedEdge e : finder.cycle()) {
                StdOut.print(e + " ");
            }
            StdOut.println();
        }

        // lub istnieje porządek topologiczny
        else {
            StdOut.println("Brak cyklu skierowanego");
        }
    }

}
