Cod sursa(job #1705476)

Utilizator elninoCristian elnino Data 20 mai 2016 17:35:49
Problema Sortare topologica Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <utility>
#include <fstream>
#include <algorithm>
#include <vector>
#include <list>
#include <stack>

struct Node {
    std::list<int> neighs;
    Node() {}
};

std::list<int> result;

void dfs(int N, int init_node, bool *visited, std::vector<Node> &graph) {
    std::stack<int> S;
    int curr_node;

    S.push(init_node);

    while(!S.empty()) {
        curr_node = S.top(); S.pop();

        if (!visited[curr_node]) {
            visited[curr_node] = true;
            for (std::list<int>::iterator neigh = graph[curr_node].neighs.begin();
                 neigh != graph[curr_node].neighs.end(); neigh++) {
                if (!visited[*neigh]) { S.push(*neigh); };
            }
        }

        result.push_back(curr_node);
    }
}

int main(int argc, char **argv) {

    std::ifstream input("sortaret.in");
    std::ofstream output("sortaret.out");
    int N, M, u, v;

    if (!input.is_open() || !output.is_open()) {
        std::cerr << "Cannot open input/output file!" << std::endl;
        std::exit(-1);
    }

    // Citire Graph
    input >> N >> M;

    std::vector<Node> graph(N + 1);

    while(input >> u >> v)
        graph[u].neighs.push_back(v);

    bool visited[N + 1];
    std::memset(visited, 0, N + 1);

    // DFS
    for (int i = 1; i <= N; i++)
        if (!visited[i])
            dfs(N, 1, visited, graph);

    for (std::list<int>::iterator it = result.begin(); it != result.end(); it++)
        output << *it << " ";

    output << std::endl;

    input.close();
    output.close();

    return 0;
}