#include <bits/stdc++.h>
using namespace std;
typedef complex<long long> point;
const long long INF = (long long)1e18;
long long dot(point a, point b) {
return (conj(a) * b).real();
}
long long cross(point a, point b) {
return (conj(a) * b).imag();
}
struct DSU {
vector<int> f;
DSU(int n) : f(n) { iota(f.begin(), f.end(), 0); }
int leader(int x) {
while (x != f[x]) {
x = f[x] = f[f[x]];
}
return x;
}
bool unionize(int x, int y) {
x = leader(x);
y = leader(y);
if (x == y) {
return false;
}
f[y] = x;
return true;
}
};
int n, m;
vector<array<long long, 5>> edges0;
point f(point axis) { // find mst on axis
vector<array<long long, 5>> edges = edges0;
for (int i = 0; i < m; i++) {
edges[i][4] = dot(axis, point(edges[i][2], edges[i][3]));
}
sort(edges.begin(), edges.end(), [&](auto x, auto y) {
return x[4] > y[4];
});
point res(0, 0);
DSU dsu(n);
for (auto [u, v, a, b, w] : edges) {
if (dsu.unionize(u, v)) {
res += point(a, b);
}
}
return res;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> m;
edges0.resize(m);
for (int i = 0; i < m; i++) {
int u, v, a, b;
cin >> u >> v >> a >> b;
edges0[i] = {u, v, -a, -b, 0};
}
point l(1, 0);
point r(0, 1);
vector<array<point, 2>> S;
function<void(point, point)> divide = [&](point l, point r) {
point lp = f(l);
point rp = f(r);
if (lp == rp) {
S.push_back({lp, l});
return;
}
point mid = (l + r);
mid /= 2;
if (cross(mid, l) == 0 || cross(mid, r) == 0) {
return;
}
point mp = f(mid);
if (mp == lp || mp == rp) {
S.push_back({lp, l});
return;
}
divide(l, mid);
divide(mid, r);
};
divide(point(500, 0), point(0, 500));
long long mn = INF;
point ans;
point key;
for (auto [p, axis] : S) {
long long prod = p.real() * p.imag();
if (prod < mn) {
ans = p;
key = axis;
mn = prod;
}
}
ans *= -1;
cout << ans.real() << " " << ans.imag() << "\n";
vector<array<long long, 5>> edges = edges0;
for (int i = 0; i < m; i++) {
edges[i][4] = dot(key, point(edges[i][2], edges[i][3]));
}
sort(edges.begin(), edges.end(), [&](auto x, auto y) {
return x[4] > y[4];
});
DSU dsu(n);
for (auto [u, v, a, b, w] : edges) {
if (dsu.unionize(u, v)) {
cout << u << " " << v << "\n";
}
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
1 ms |
340 KB |
Output is correct |
7 |
Correct |
3 ms |
468 KB |
Output is correct |
8 |
Correct |
6 ms |
1108 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Incorrect |
1 ms |
212 KB |
Output isn't correct |
11 |
Correct |
1 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
11 ms |
340 KB |
Output is correct |
15 |
Incorrect |
7 ms |
340 KB |
Output isn't correct |
16 |
Correct |
190 ms |
536 KB |
Output is correct |
17 |
Correct |
182 ms |
536 KB |
Output is correct |
18 |
Correct |
197 ms |
540 KB |
Output is correct |
19 |
Correct |
1324 ms |
1108 KB |
Output is correct |
20 |
Correct |
1407 ms |
1108 KB |
Output is correct |