QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#528650 | #1825. The King's Guards | PetroTarnavskyi# | WA | 497ms | 5192kb | C++20 | 3.2kb | 2024-08-23 18:37:55 | 2024-08-23 18:37:55 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for(int i = (a); i < (b); i++)
#define RFOR(i, a, b) for(int i = (a) - 1; i >= (b); i--)
#define SZ(a) int(a.size())
#define ALL(a) a.begin(), a.end()
#define PB push_back
#define MP make_pair
#define F first
#define S second
typedef long long LL;
typedef vector<int> VI;
typedef pair<int, int> PII;
typedef double db;
mt19937 rng;
struct Graph
{
int szL, szR;
vector<VI> g;
VI mateForR, mateForL, usedL;
void init(int L, int R)
{
szL = L, szR = R;
g.resize(szL);
mateForL.resize(szL);
usedL.resize(szL);
mateForR.resize(szR);
}
void addEdge(int from, int to)
{
assert(0 <= from && from < szL);
assert(0 <= to && to < szR);
g[from].PB(to);
}
int iter;
bool kuhn(int v)
{
if (usedL[v] == iter) return false;
usedL[v] = iter;
shuffle(ALL(g[v]), rng);
for (int to : g[v])
{
if (mateForR[to] == -1)
{
mateForR[to] = v;
mateForL[v] = to;
return true;
}
}
for (int to : g[v])
{
if (kuhn(mateForR[to]))
{
mateForR[to] = v;
mateForL[v] = to;
return true;
}
}
return false;
}
int doKuhn()
{
fill(ALL(mateForR), -1);
fill(ALL(mateForL), -1);
fill(ALL(usedL), -1);
int res = 0;
iter = 0;
while (true)
{
iter++;
bool ok = false;
FOR (v, 0, szL)
{
if (mateForL[v] == -1)
{
if (kuhn(v))
{
ok = true;
res++;
}
}
}
if (!ok) break;
}
return res;
}
};
struct DSU
{
int n;
VI p, sz;
void init(int _n)
{
n = _n;
p.resize(n);
iota(ALL(p), 0);
sz.assign(n, 1);
}
int find(int v)
{
if (v == p[v])
return v;
return p[v] = find(p[v]);
}
bool unite(int u, int v)
{
u = find(u);
v = find(v);
if (u == v)
return false;
if (sz[u] > sz[v])
swap(u, v);
p[u] = v;
sz[v] += sz[u];
return true;
}
};
int n, r, g;
bool check(vector<pair<int, PII>>& e, vector<VI>& gg)
{
DSU d;
d.init(n);
for (auto [c, p] : e)
d.unite(p.F, p.S);
map<int, int> m;
FOR (i, 0, n)
{
int c = d.find(i);
if (!m.count(c))
m[c] = SZ(m);
}
Graph G;
G.init(SZ(m), g);
FOR (i, 0, g)
{
for (auto v : gg[i])
G.addEdge(m[d.find(v)], i);
}
int res = G.doKuhn();
return res == SZ(m);
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cin >> n >> r >> g;
vector<pair<int, PII>> edges;
FOR (i, 0, r)
{
int a, b, c;
cin >> a >> b >> c;
a--, b--;
edges.PB({c, {a, b}});
}
vector<VI> gg(g);
FOR (i, 0, g)
{
int k;
cin >> k;
FOR (j, 0, k)
{
int v;
cin >> v;
gg[i].PB(v - 1);
}
}
sort(ALL(edges));
vector<pair<int, PII>> e;
int cost = 0;
DSU d;
d.init(n);
for (auto [c, p] : edges)
{
auto [u, v] = p;
if (d.unite(u, v))
{
e.PB({c, p});
cost += c;
}
}
if (!check(e, gg))
{
cout << -1 << '\n';
return 0;
}
int sz = SZ(e);
RFOR (i, sz, 0)
{
vector<pair<int, PII>> ne;
FOR (j, 0, SZ(e))
{
if (i != j)
ne.PB(e[j]);
}
if (check(ne, gg))
{
cost -= e[i].F;
e = ne;
}
}
cout << cost << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3608kb
input:
5 6 2 1 2 1 1 3 4 2 4 2 2 5 5 3 4 7 4 5 3 2 1 2 2 2 4
output:
8
result:
ok answer is '8'
Test #2:
score: 0
Accepted
time: 0ms
memory: 3620kb
input:
10 19 6 1 5 761 6 8 606 3 9 648 2 4 115 5 8 814 1 2 712 4 10 13 5 10 797 3 4 956 1 7 73 5 7 192 2 7 110 5 9 302 3 6 120 6 9 494 1 3 668 3 7 966 6 10 974 3 8 41 2 10 5 3 6 4 3 2 1 7 2 10 8 3 10 7 8 2 2 1
output:
429
result:
ok answer is '429'
Test #3:
score: 0
Accepted
time: 0ms
memory: 3612kb
input:
10 43 3 1 3 656 2 6 856 4 10 99 5 6 900 2 7 766 4 7 582 2 8 135 5 7 831 3 5 12 3 10 789 1 8 66 4 9 390 1 7 238 6 7 960 1 4 624 3 9 602 7 10 366 5 8 526 2 9 561 6 10 722 2 5 904 3 4 35 1 9 768 5 9 457 6 8 61 4 6 192 4 5 96 5 10 747 8 9 611 7 8 953 3 8 449 2 4 228 1 6 197 9 10 160 3 6 869 1 2 785 4 8 ...
output:
526
result:
ok answer is '526'
Test #4:
score: 0
Accepted
time: 4ms
memory: 3704kb
input:
277 9038 1 226 260 740 44 226 376 151 263 611 67 269 241 120 181 677 259 271 782 37 52 310 48 152 452 168 266 823 85 234 100 46 201 738 129 153 301 69 147 434 13 72 764 13 234 316 171 222 398 214 255 21 112 158 430 20 118 407 45 152 971 205 214 272 221 275 362 198 268 472 117 176 207 31 75 652 139 1...
output:
5375
result:
ok answer is '5375'
Test #5:
score: 0
Accepted
time: 26ms
memory: 3788kb
input:
297 27966 132 15 197 980 226 259 950 161 168 142 118 176 834 157 221 806 24 210 432 212 242 838 110 166 177 78 170 801 52 166 3 89 213 448 45 170 626 250 251 268 93 222 679 7 128 839 5 7 320 132 191 1 192 295 717 36 231 542 162 175 508 173 178 458 211 272 926 46 168 145 19 150 805 165 262 198 50 179...
output:
775
result:
ok answer is '775'
Test #6:
score: 0
Accepted
time: 0ms
memory: 3548kb
input:
7 7 4 1 3 7 1 4 6 2 3 5 2 4 6 4 5 10 4 6 10 4 7 10 5 4 3 2 7 5 6 5 2 6 7 4 1 2 2 3 6 7 2 5 3 1 6
output:
17
result:
ok answer is '17'
Test #7:
score: 0
Accepted
time: 497ms
memory: 4600kb
input:
300 44850 299 1 2 1 1 3 1 1 4 1 1 5 1 1 6 1 1 7 1 1 8 1 1 9 1 1 10 1 1 11 1 1 12 1 1 13 1 1 14 1 1 15 1 1 16 1 1 17 1 1 18 1 1 19 1 1 20 1 1 21 1 1 22 1 1 23 1 1 24 1 1 25 1 1 26 1 1 27 1 1 28 1 1 29 1 1 30 1 1 31 1 1 32 1 1 33 1 1 34 1 1 35 1 1 36 1 1 37 1 1 38 1 1 39 1 1 40 1 1 41 1 1 42 1 1 43 1 ...
output:
1000
result:
ok answer is '1000'
Test #8:
score: 0
Accepted
time: 326ms
memory: 5192kb
input:
300 44850 299 1 2 1 1 3 1 1 4 1 1 5 1 1 6 1 1 7 1 1 8 1 1 9 1 1 10 1 1 11 1 1 12 1 1 13 1 1 14 1 1 15 1 1 16 1 1 17 1 1 18 1 1 19 1 1 20 1 1 21 1 1 22 1 1 23 1 1 24 1 1 25 1 1 26 1 1 27 1 1 28 1 1 29 1 1 30 1 1 31 1 1 32 1 1 33 1 1 34 1 1 35 1 1 36 1 1 37 1 1 38 1 1 39 1 1 40 1 1 41 1 1 42 1 1 43 1 ...
output:
1000
result:
ok answer is '1000'
Test #9:
score: 0
Accepted
time: 118ms
memory: 4124kb
input:
300 44850 150 1 2 4 1 3 4 1 4 4 1 5 3 1 6 10 1 7 4 1 8 8 1 9 5 1 10 4 1 11 9 1 12 9 1 13 1 1 14 3 1 15 5 1 16 7 1 17 5 1 18 2 1 19 6 1 20 4 1 21 10 1 22 5 1 23 7 1 24 2 1 25 2 1 26 4 1 27 4 1 28 5 1 29 8 1 30 10 1 31 9 1 32 1 1 33 7 1 34 5 1 35 4 1 36 6 1 37 8 1 38 1 1 39 2 1 40 1 1 41 1 1 42 1 1 43...
output:
249
result:
ok answer is '249'
Test #10:
score: 0
Accepted
time: 119ms
memory: 4060kb
input:
300 44850 150 1 2 70 1 3 26 1 4 76 1 5 74 1 6 98 1 7 72 1 8 66 1 9 25 1 10 36 1 11 57 1 12 8 1 13 88 1 14 98 1 15 33 1 16 85 1 17 56 1 18 16 1 19 62 1 20 41 1 21 81 1 22 18 1 23 15 1 24 69 1 25 11 1 26 29 1 27 62 1 28 64 1 29 41 1 30 92 1 31 29 1 32 99 1 33 40 1 34 30 1 35 23 1 36 49 1 37 6 1 38 84 ...
output:
1299
result:
ok answer is '1299'
Test #11:
score: 0
Accepted
time: 160ms
memory: 4268kb
input:
300 44850 150 1 2 3 1 3 2 1 4 20 1 5 77 1 6 77 1 7 23 1 8 39 1 9 57 1 10 83 1 11 60 1 12 6 1 13 78 1 14 64 1 15 62 1 16 41 1 17 88 1 18 77 1 19 58 1 20 39 1 21 18 1 22 43 1 23 26 1 24 40 1 25 61 1 26 23 1 27 6 1 28 47 1 29 100 1 30 59 1 31 92 1 32 60 1 33 13 1 34 57 1 35 55 1 36 99 1 37 77 1 38 63 1...
output:
1310
result:
ok answer is '1310'
Test #12:
score: 0
Accepted
time: 0ms
memory: 4024kb
input:
300 44551 299 1 2 1 1 3 1 1 4 1 1 5 1 1 6 1 1 7 1 1 8 1 1 9 1 1 10 1 1 11 1 1 12 1 1 13 1 1 14 1 1 15 1 1 16 1 1 17 1 1 18 1 1 19 1 1 20 1 1 21 1 1 22 1 1 23 1 1 24 1 1 25 1 1 26 1 1 27 1 1 28 1 1 29 1 1 30 1 1 31 1 1 32 1 1 33 1 1 34 1 1 35 1 1 36 1 1 37 1 1 38 1 1 39 1 1 40 1 1 41 1 1 42 1 1 43 1 ...
output:
-1
result:
ok answer is '-1'
Test #13:
score: 0
Accepted
time: 36ms
memory: 4072kb
input:
300 44552 299 1 2 1 1 3 1 1 4 1 1 5 1 1 6 1 1 7 1 1 8 1 1 9 1 1 10 1 1 11 1 1 12 1 1 13 1 1 14 1 1 15 1 1 16 1 1 17 1 1 18 1 1 19 1 1 20 1 1 21 1 1 22 1 1 23 1 1 24 1 1 25 1 1 26 1 1 27 1 1 28 1 1 29 1 1 30 1 1 31 1 1 32 1 1 33 1 1 34 1 1 35 1 1 36 1 1 37 1 1 38 1 1 39 1 1 40 1 1 41 1 1 42 1 1 43 1 ...
output:
2
result:
ok answer is '2'
Test #14:
score: -100
Wrong Answer
time: 36ms
memory: 4060kb
input:
300 44552 300 1 2 1 1 3 1 1 4 1 1 5 1 1 6 1 1 7 1 1 8 1 1 9 1 1 10 1 1 11 1 1 12 1 1 13 1 1 14 1 1 15 1 1 16 1 1 17 1 1 18 1 1 19 1 1 20 1 1 21 1 1 22 1 1 23 1 1 24 1 1 25 1 1 26 1 1 27 1 1 28 1 1 29 1 1 30 1 1 31 1 1 32 1 1 33 1 1 34 1 1 35 1 1 36 1 1 37 1 1 38 1 1 39 1 1 40 1 1 41 1 1 42 1 1 43 1 ...
output:
2
result:
wrong answer expected '-1', found '2'