QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#387856 | #7871. 命运 | valeriu# | 15 | 72ms | 27236kb | C++20 | 5.9kb | 2024-04-12 22:04:51 | 2024-07-04 03:36:27 |
Judging History
answer
#include <bits/stdc++.h>
#define all(x) (x).begin(),(x).end()
using namespace std;
using ll = long long;
using ld = long double;
#define int ll
#define sz(x) ((int)(x).size())
using pii = pair<int,int>;
using tii = tuple<int,int,int>;
const int nmax = 5e4 + 5, inf = 1e9 + 6;
vector<pii> g[nmax];
int color[nmax];
void fill(int node, int C) {
if(color[node] != 0) return;
color[node] = C;
//cerr << node << ' ' << C << '\n';
for(auto [x, w] : g[node]) fill(x, C);
}
namespace DSU {
int dsu[nmax], links[nmax];
int cc, totalK;
ll stdcost, supra;
vector<tii> stundo;
void init(int n) {
cc = n;
stdcost = 0;
for(int i = 1; i <= n; i++) dsu[i] = -1, links[i] = 0;
return;
}
int f(int x) { return dsu[x] < 0? x : f(dsu[x]); }
bool unite(int x, int y) {
x = f(x);
y = f(y);
if(x == y) {
stundo.emplace_back(x, y, 0);
return 0;
}
if(-dsu[x] < -dsu[y]) swap(x, y);
stundo.emplace_back(x, y, dsu[y]);
dsu[x] += dsu[y];
dsu[y] = x;
links[x] += links[y];
cc--;
return 1;
}
void addlink(int node, int aggr) {
links[node] += aggr;
if(dsu[node] < 0) { totalK += aggr; return; }
addlink(dsu[node], aggr);
}
bool undo() {
auto [x, y, A] = stundo.back();
stundo.pop_back();
if(x == y) return 0;
dsu[y] = A;
dsu[x] -= A;
links[x] -= links[y];
cc++;
return links[x] == 0 || links[y] == 0;
}
}
namespace PQ_Undo {
set<pii> bypri;
vector<pair<pii, int>> st;
vector<int> pos_inst, atrpri;
vector<pii> atrupd;
void push(pii upd, int pri) {
atrpri.emplace_back(pri);
atrupd.emplace_back(upd);
DSU::unite(upd.first, upd.second);
st.emplace_back(make_pair(upd, sz(pos_inst)));
pos_inst.emplace_back(sz(st) - 1);
bypri.emplace(make_pair(pri, sz(pos_inst) - 1));
}
bool pop() {
int k = sz(st), last_priority;
vector<pair<pii, int>> big, small;
auto it = bypri.begin();
for(int qt = 1, lol = 0; !lol || qt < (sz(st) + 1) / 2; lol = 1, qt++) {
int individ = it -> second;
k = min(pos_inst[individ], k);
/* if(qt > 1) */ big.emplace_back(make_pair(atrupd[individ], individ));
last_priority = atrpri[individ];
if(qt * 2 > sz(st) - k) break;
it++;
}
bypri.erase(bypri.begin());
//cerr << sz(st) << ' ' << sz(DSU::stundo) << '\n';
while(sz(st) > k) {
auto [upd, individ] = st.back();
st.pop_back();
DSU::undo();
if(atrpri[individ] > last_priority) small.emplace_back(make_pair(upd, individ));
}
copy(rbegin(big), rend(big), back_inserter(small));
for(auto [upd, individ] : small) {
DSU::unite(upd.first, upd.second);
st.emplace_back(make_pair(upd, individ));
pos_inst[individ] = sz(st) - 1;
}
assert(atrpri[st.back().second] != inf);
st.pop_back();
return DSU::undo();
}
}
signed main() {
cin.tie(0) -> sync_with_stdio(0);
int n, m, k, s;
cin >> n >> m >> k >> s;
vector<tii> edgs;
for(int i = 0, x, y, w; i < m; i++) {
cin >> x >> y >> w;
if(y == s) swap(x, y);
else if(x != s && x > y) swap(x, y);
edgs.emplace_back(x, y, w);
}
pii prv = {-1, -1};
sort(all(edgs));
vector<tii> nvedgs;
for(auto [x, y, w] : edgs) {
if(prv == make_pair(x, y) && x == s) continue;
g[x].emplace_back(y, w);
g[y].emplace_back(x, w);
prv = pii{x, y};
nvedgs.emplace_back(x, y, w);
}
edgs = move(nvedgs);
{
color[s] = -1;
int C = 1;
for(int i = 1; i <= n; i++) {
if(color[i] == 0) fill(i, C++);
}
C--;
if(sz(g[s]) < k) {
cout << "nonnondayo\n";
return 0;
}
if(!([&]() {
vector<int> found_colors(C, 0);
for(auto [x, w] : g[s]) found_colors[color[x] - 1] = 1;
if((accumulate(all(found_colors), 0) != C) || C > k) return false;
return true;
}())) {
cout << "nonnondayo\n";
return 0;
}
}
DSU::init(n);
DSU::cc--;
sort(all(edgs), [&](auto a, auto b) { return get<2>(a) < get<2>(b); });
for(auto [x, w] : g[s]) {
DSU::addlink(x, 1);
}
vector<tii> keeper;
int II = sz(edgs) + 2;
for(auto [a, b, w] : edgs) {
//cout << a << ' ' << b << ' ' << w << '\n';
if(a == s) continue;
PQ_Undo::push(pii{a, b}, II);
II--;
}
for(int i = sz(edgs) - 1; i >= 0; i--) {
auto [a, b, w] = edgs[i];
if(a != s) {
if(PQ_Undo::pop() || DSU::cc > k) {
keeper.emplace_back(edgs[i]);
PQ_Undo::push(pii{a, b}, inf);
//while(sz(DSU::stundo)) DSU::undo();
//for(auto [u, v, T] : keeper) if(u != s) DSU::unite(u, v);
//for(auto [u, v, T] : edgs | views::take(i)) if(u != s) DSU::unite(u, v);
}
}
else {
if(DSU::links[DSU::f(b)] == 1 || DSU::totalK == k)
//cerr << a << ' ' << b << ' ' << w << '\n',
keeper.emplace_back(edgs[i]);
//k--;
else
DSU::addlink(b, -1);
}
}
//cerr << sz(keeper) << '\n';
ll total = 0;
for(auto [a, b, w] : keeper) {
total += w;
//cout << a << ' ' << b << '\n';
}
cout << total << '\n';
}
/**
Istenem! Nu poate fi real.
-- Surse verificate
*/
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 5
Accepted
Test #1:
score: 5
Accepted
time: 62ms
memory: 22764kb
input:
49277 49276 1 49277 1 2 2000 2 3 3000 2 4 4000 3 5 5000 3 6 6000 4 7 7000 1 8 8000 7 9 9000 1 10 10000 5 11 11000 4 12 12000 3 13 13000 13 14 14000 1 15 15000 7 16 16000 11 17 17000 2 18 18000 10 19 19000 6 20 20000 4 21 21000 17 22 22000 1 23 23000 14 24 24000 4 25 25000 16 26 26000 15 27 27000 9 2...
output:
1214136002000
result:
ok single line: '1214136002000'
Test #2:
score: 0
Accepted
time: 17ms
memory: 9048kb
input:
49324 49323 1 49324 1 2 2 2 3 3 2 4 4 3 5 5 4 6 6 6 7 7 2 8 8 2 9 9 4 10 10 6 11 11 9 12 12 4 13 13 8 14 14 8 15 15 7 16 16 11 17 17 15 18 18 2 19 19 10 20 20 19 21 21 14 22 22 16 23 23 20 24 24 23 25 25 3 26 26 2 27 27 8 28 28 11 29 29 20 30 30 15 31 31 16 32 32 19 33 33 29 34 34 5 35 35 21 36 36 1...
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Test #3:
score: 0
Accepted
time: 16ms
memory: 8980kb
input:
49423 49422 1 1 1 2 2 2 3 3 1 4 4 3 5 5 2 6 6 1 7 7 1 8 8 7 9 9 3 10 10 3 11 11 5 12 12 3 13 13 9 14 14 10 15 15 13 16 16 5 17 17 9 18 18 12 19 19 17 20 20 4 21 21 5 22 22 12 23 23 21 24 24 21 25 25 11 26 26 17 27 27 21 28 28 18 29 29 17 30 30 2 31 31 21 32 32 28 33 33 17 34 34 31 35 35 11 36 36 8 3...
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Test #4:
score: 0
Accepted
time: 13ms
memory: 9240kb
input:
49501 49500 49501 49501 1 2 2 1 3 3 3 4 4 2 5 5 5 6 6 1 7 7 6 8 8 5 9 9 6 10 10 5 11 11 2 12 12 12 13 13 13 14 14 8 15 15 13 16 16 5 17 17 9 18 18 9 19 19 9 20 20 14 21 21 18 22 22 16 23 23 7 24 24 1 25 25 7 26 26 1 27 27 15 28 28 22 29 29 20 30 30 12 31 31 4 32 32 16 33 33 22 34 34 11 35 35 27 36 3...
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Subtask #2:
score: 10
Accepted
Test #5:
score: 10
Accepted
time: 1ms
memory: 3556kb
input:
21 20 2 21 1 2 18581 1 3 9620 2 4 4362 1 5 7702 5 6 17435 4 7 11679 3 8 14832 1 9 15073 2 10 6595 5 11 19676 11 12 16943 12 13 5268 5 14 20262 8 15 8192 7 16 12340 7 17 1437 7 18 3064 2 19 10437 12 20 2882 19 21 13483
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Test #6:
score: 0
Accepted
time: 1ms
memory: 3624kb
input:
10 20 4 3 1 2 18247 2 3 9041 2 4 4031 2 5 7363 3 6 17172 1 7 11014 6 8 14781 8 9 15347 8 10 6598 5 9 19331 3 10 16523 10 6 5732 2 3 20357 6 9 8827 10 3 12864 6 3 1551 5 6 3932 3 1 10223 9 3 2296 8 5 13620
output:
54418
result:
ok single line: '54418'
Test #7:
score: 0
Accepted
time: 1ms
memory: 3552kb
input:
10 20 6 10 1 2 18401 2 3 9843 3 4 4219 4 5 7552 4 6 17751 4 7 11318 5 8 14774 8 9 15541 5 10 6928 6 10 19860 10 5 16699 5 8 5937 5 2 20611 1 8 8077 10 1 12386 9 4 1663 9 10 3910 1 9 10401 7 10 2383 2 9 13681
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Test #8:
score: 0
Accepted
time: 1ms
memory: 3808kb
input:
10 20 1 10 1 2 18853 2 3 9034 3 4 4069 3 5 7743 3 6 17122 6 7 11715 2 8 14814 1 9 15011 7 10 6248 6 8 19400 7 3 16354 6 8 5249 7 8 20701 5 10 8079 10 5 12570 7 10 1006 8 3 3814 5 10 10753 5 9 2310 8 6 13123
output:
59951
result:
ok single line: '59951'
Test #9:
score: 0
Accepted
time: 1ms
memory: 3636kb
input:
10 11 3 1 1 2 9407 1 3 7005 1 4 5453 1 5 4585 1 6 8278 1 7 6332 1 8 1415 1 9 3809 1 10 10519 2 6 2507 10 6 11580
output:
nonnondayo
result:
ok single line: 'nonnondayo'
Subtask #3:
score: 0
Wrong Answer
Dependency #2:
100%
Accepted
Test #10:
score: 0
Wrong Answer
time: 3ms
memory: 4432kb
input:
1000 3000 100 10 1 2 2270918 1 3 871400 2 4 1242131 3 5 2307909 5 6 1391239 3 7 1431327 7 8 2501929 5 9 2377716 8 10 612146 5 11 991199 11 12 1810465 10 13 1094558 10 14 2605381 8 15 872336 10 16 2092297 5 17 619719 14 18 1161002 5 19 1828768 10 20 837691 2 21 1787203 3 22 396276 21 23 1371664 16 24...
output:
647499805
result:
wrong answer 1st lines differ - expected: '645739778', found: '647499805'
Subtask #4:
score: 0
Wrong Answer
Dependency #1:
100%
Accepted
Test #14:
score: 0
Wrong Answer
time: 67ms
memory: 23000kb
input:
49997 54855 3516 1 3 2 123052288 4 2 96660489 5 4 21916339 6 4 110026562 7 2 37432698 8 4 38560777 9 5 83209343 10 9 80352789 11 2 88956732 12 7 65449905 13 2 38239108 14 5 90154247 15 4 30810782 16 13 96492679 17 14 112886179 18 9 87190321 19 5 91181643 20 3 107304129 21 15 101439791 22 19 12060197...
output:
2690063795693
result:
wrong answer 1st lines differ - expected: '2688197428747', found: '2690063795693'
Subtask #5:
score: 0
Wrong Answer
Test #21:
score: 0
Wrong Answer
time: 72ms
memory: 27236kb
input:
9992 99999 500 1 1 2 96661121 1 3 21915940 1 4 110026320 1 5 37433129 1 6 38560320 1 7 83209024 1 8 80352054 1 9 88957672 1 10 65449874 1 11 38239186 1 12 90153728 1 13 30810974 1 14 96493404 1 15 112886259 1 16 87190053 1 17 91182163 1 18 107303768 1 19 101439823 1 20 120601875 1 21 122197599 1 22 ...
output:
113113115376
result:
wrong answer 1st lines differ - expected: '112890891818', found: '113113115376'
Subtask #6:
score: 0
Skipped
Dependency #3:
0%
Subtask #7:
score: 0
Skipped
Dependency #1:
100%
Accepted
Dependency #2:
100%
Accepted
Dependency #3:
0%