Submission #1008364

# Submission time Handle Problem Language Result Execution time Memory
1008364 2024-06-26T10:01:59 Z overwatch9 Swapping Cities (APIO20_swap) C++17
Compilation error
0 ms 0 KB
#include "swap.h"

#include <bits/stdc++.h>
using namespace std;

vector <vector <pair <int, int>>> adj;
vector <array <int, 3>> edges;
int n, m;
bool comp(array <int, 3> a, array <int, 3> b) {
	return a[2] < b[2];
}
void init(int N, int M, vector<int> U, vector<int> V, vector<int> W) {
	// ios_base::sync_with_stdio(false);
	// cin.tie(0);
	// cout.tie(0);
    adj.resize(N+1);
	n = N;
	m = M;
    for (int i = 0; i < M; i++) {
        adj[U[i]].push_back({V[i], W[i]});
		adj[V[i]].push_back({U[i], W[i]});
		edges.push_back({U[i], V[i], W[i]});
	}
	sort(edges.begin(), edges.end(), comp);
}
struct DSU {
	vector <int> link, sz;
	DSU (int s) {
		link = sz = vector <int> (s+1);
		for (int i = 1; i <= s; i++) {
			link[i] = i;
			sz[i] = 1;
		}
	}
	int head(int x) {
		if (link[x] != x)
			return link[x] = head(link[x]);
		return x;
	}
	bool same(int a, int b) {
		return head(a) == head(b);
	}
	void unite(int a, int b) {
		a = head(a);
		b = head(b);
		if (sz[a] < sz[b])
			swap(a, b);
		sz[a] += sz[b];
		link[b] = a;
	}
};
int getMinimumFuelCapacity(int x, int y) {
	DSU dsu(n+1);
	int ans = -1;
	for (int i = 0; i < m; i++) {
		int a = edges[i][0], b = edges[i][1];
		if (!dsu.same(a, b))
			dsu.unite(a, b);
		else {
			if (dsu.same(a, x)) {
				if (dsu.same(x, y)) {
					ans = edges[i][2];
				}
			}
		}
	}
	return ans;
}
int main() {
	int N, M;
	cin >> N >> M;
	vector <int> u(M), v(M), w(M);
	for (int i = 0; i < M; i++)
		cin >> u[i] >> v[i] >> w[i];
	init(N, M, u, v, w);
	int q;
	cin >> q;
	while (q--) {
		int x, y;
		cin >> x >> y;
		cout << getMinimumFuelCapacity(x, y) << '\n';
	}
}

Compilation message

/usr/bin/ld: /tmp/ccOyHvz6.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccoI7ap7.o:swap.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status