Distinct Routes

#include <algorithm>
#include <climits>
#include <cmath>
#include <iostream>
#include <queue>
#include <vector>

// Debug printing
#ifdef DEBUG
#define deb(fmt, args...) printf("DEBUG: %d: " fmt, __LINE__, ##args)
#else
#define deb(fmt, args...)
#endif

using namespace std;
using ll = long long;

void print_array(vector<int> in, const string title = "Vector")
{
	cout << title << " [\n";
	for (const auto &el : in) {
		cout << el << " ";
	}
	cout << "\n] END\n";
}

void print_matrix(vector<vector<int> > in, const string title = "Matrix")
{
	cout << title << " [\n";
	for (unsigned int i = 0; i < in.size(); i++) {
		cout << "row " << i << ": ";
		for (unsigned int j = 0; j < in[i].size(); j++) {
			cout << in[i][j] << " ";
		}
		cout << "\n";
	}
	cout << "] END\n";
}

int target;
int found = true;
vector<int> path;
bool visited[500];

void dijkstra(vector<vector<int> > &adj, vector<vector<int> > &edges, int n,
	      vector<int> &path)
{
	priority_queue<pair<ll, ll> > q;
	vector<int> distance(n + 1, INT_MAX);
	vector<int> predecessor(n + 1, 0);
	vector<bool> processed(n + 1, false);

	distance[1] = 0;
	q.push({ 0, 1 });
	while (!q.empty()) {
		ll a = q.top().second;
		q.pop();
		if (processed[a])
			continue;
		processed[a] = true;
		for (auto u : adj[a]) {
			vector<int> edge = edges[u];
            if (edge[2] < 1) continue;
			ll b = edge[1], w = 1;
			if (distance[a] + w < distance[b]) {
				distance[b] = distance[a] + w;
				predecessor[b] = u;
				q.push({ -distance[b], b });
			}
		}
	}

	// find shortest path
    if (predecessor[n] == 0) {
        found = false;
        return;
    }
    found = true;
    int v = n;
	int e = predecessor[v];
    int index = 0;
	while (v != 1) {
		e = predecessor[v];
		path.push_back(e);
		v = edges[e][0];
        index++;
	}
    reverse(path.begin(), path.end());
}

void dfs(int s, vector<vector<int> > &adj, vector<vector<int> > &edges,
	 vector<int> &path)
{
	if (found) {
		return;
	}
	if (visited[s]) {
		return;
	};
	visited[s] = true;
	if (s == target) {
		found = true;
		return;
	}
 
	for (auto u : adj[s]) {
        // ll a = edges[u][0];
        ll b = edges[u][1];
        ll c = edges[u][2];
        // deb("s: %d, a: %d, b: %d, c: %d\n", s, a, b, c);
        if (c > 0) {
            path.push_back(u);
            dfs(b, adj, edges, path);
            if (found) return;
            path.pop_back();
        }
	}
}

int main(int argc, char *argv[])
{
	int n, m;
	cin >> n >> m;
	target = n;

	// each edge is of form (a, b, c)
	vector<vector<int> > edges(2 * m + 2, vector<int>(3));
	for (int i = 1; i < 2 * m; i += 2) {
		cin >> edges[i][0] >> edges[i][1];
        edges[i + 1][0] = edges[i][1];
        edges[i + 1][1] = edges[i][0];
		// cin >> edges[i + 1][0] >> edges[i][1];
        edges[i][2] = 1;
        edges[i + 1][2] = 0;
	}

    // deb("paskaa\n");

	// adjacent structure contains the edge indices that are adjacent to the
	// node
	vector<vector<int> > adj(n + 1, vector<int>());
	for (int j = 1; j < 2*(m + 1); j++) {
		adj[edges[j][0]].push_back(j);
	}

    vector<vector<int>> paths;
    while (found) {
        found = false;
		for (int i = 0; i < n + 1; i++)
			visited[i] = false;

		path.clear();
        dfs(1, adj, edges, path);
        // dijkstra(adj, edges, n, path);
        if (found) {
            paths.push_back(path);
        }
        for (auto e : path) {
            if (e % 2 == 0) {
                edges[e][2] = 0;
                edges[e + 1][2] = 1;
            } else {
                edges[e][2] = 0;
                edges[e - 1][2] = 1;
            }
        }
    }

    // print answer
    cout << paths.size() << "\n";
    for (auto path : paths) {
        cout << path.size() + 1 << "\n";
        for (auto e : path) {
            cout << edges[e][0] << " ";
            edges[e][2] -= 1;
        }
        cout << n << "\n";
    }

    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: WRONG ANSWER