QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#758185 | #8136. Rebellious Edge | raspa | TL | 3ms | 35736kb | C++17 | 2.3kb | 2024-11-17 16:34:20 | 2024-11-17 16:34:22 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
//#pragma GCC optimize("O3,unroll-loops")
#define migmig ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define F first
#define S second
#define SZ(x) ll((x).size())
#define all(x) (x).begin(), (x).end()
#define cl clear
#define endl '\n'
#define deb(x) cerr << #x << " = " << x << '\n'
#define dokme(x) {cout << x << endl; return;}
#define wtf cout << "[ahaaaaaaaaaaaaaaaa]" << endl;
const int MAXN = 1e6 + 10;
const int MOD = 1e9 + 7; //998244353
const ll INF = 1e18;
const int inf = 1e9 + 90;
const int maxLG = 20;
const int SQ = 350;
ll n, m, dist[MAXN];
vector<pll> G[MAXN];
bool mark[MAXN];
set<pii> se;
ll MST = 0;
ll cnt = 0;
void add(ll v) {
for (auto [u, w] : G[v]) {
if (!mark[u]) {
se.erase({dist[u], u});
se.insert({(dist[u] = w), u});
}
}
}
void prim() {
dist[1] = 0;
add(1);
se.insert({0, 1});
for (ll i = 1; i <= n; i++) {
if (se.empty()) {
return;
}
auto tmp = *(se.begin());
ll v = tmp.S;
MST += tmp.F;
cnt++;
se.erase(tmp);
add(v);
mark[v] = 1;
}
}
void reset() {
memset(dist, 63, sizeof dist);
for (ll i = 1; i <= n; i++)
G[i].cl();
memset(dist, 63, sizeof dist);
se.cl();
memset(mark, false, sizeof mark);
MST = 0;
}
int main() {
migmig;
ll T;
cin >> T;
while (T--) {
cin >> n >> m;
reset();
ll b, tmp;
vector<pair<pll, ll>> edg;
for (ll i = 1; i <= m; i++) {
ll v, u, w;
cin >> v >> u >> w;
if (v > u) {
b = u;
tmp = w;
}
edg.pb({{v, u}, w});
}
for (auto x : edg) {
ll v = x.F.F, u = x.F.S, w = x.S;
if (v != b && u != b)
G[v].pb({u, w});
}
prim();
// deb(cnt);
ll res1 = MST + tmp;
if (cnt > n) {
res1 = inf;
}
MST = 0;
reset();
ll c;
for (auto x : edg) {
ll v = x.F.F, u = x.F.S, w = x.S;
if (v < u)
G[v].pb({u, w});
}
cnt = 0;
prim();
ll res2 = MST;
// deb(cnt);
if (cnt > n) {
res2 = inf;
}
cout << min(res1, res2) << endl;
reset();
// deb(res1 << ' ' << res2);
}
}
/*
5 6
1 2 4
1 3 2
2 3 0
3 4 1
4 5 1
5 2 1
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 3ms
memory: 35736kb
input:
3 5 6 1 2 4 1 3 2 2 3 0 3 4 1 4 5 1 5 2 1 4 4 1 2 4 1 3 6 1 4 8 4 2 1000000 3 3 1 2 100 2 1 10 2 3 1000
output:
5 18 1100
result:
ok 3 number(s): "5 18 1100"
Test #2:
score: -100
Time Limit Exceeded
input:
50000 4 5 2 4 998973548 2 3 501271695 4 1 778395982 1 4 32454891 1 2 757288934 4 5 1 4 720856669 2 3 665098951 3 4 407461517 2 1 866914401 1 2 457859826 4 5 1 2 75288664 1 4 624893594 3 2 458973866 2 4 769074655 2 3 834773751 4 5 2 3 237060634 2 4 297307882 3 4 200383586 1 2 42856502 3 1 16574713 4 ...
output:
778395982 1000000090 1000000090 480300722 1000000090 1000000090 1000000090 858095911 1000000090 793861088 635606317 1000000090 612891589 1000000090 622081756 1000000090 1000000090 93634961 960978736 984886788 1000000090 1000000090 1000000090 1000000090 1000000090 977157479 1000000090 654475526 10000...