QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#605058 | #8759. 小班课 | UESTC_NLNS | TL | 106ms | 27856kb | C++14 | 4.3kb | 2024-10-02 15:16:14 | 2024-10-02 15:16:15 |
Judging History
answer
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int N = 505;
vector<int> g[N];
int c[N];
int b[N];
int n, m;
vector<int> get_ans() {
vector<int> b1(m + 1);
for (int i = 1; i <= m; ++i) b1[i] = b[i];
queue<int> q;
vector<int> ans;
vector<int> cur(n + 1, 0);
vector<vector<int>> w(m + 1);
ans.reserve(n);
auto get = [&](int i) {
return g[i][cur[i]];
};
for (int i = 1; i <= m; ++i) {
if (b1[i] == 0) q.push(i);
}
for (int i = 1; i <= n; ++i) {
if (c[i] == 0) continue;
int w1 = g[i][cur[i]];
if (w1 == c[i]) {
// assert(b1[w1] > 0);
b1[w1]--;
if (b1[w1] == 0) q.push(w1);
ans.push_back(i);
} else {
w[w1].push_back(i);
}
}
while (q.size()) {
int f = q.front();
q.pop();
for (const auto u : w[f]) {
while (b1[get(u)] == 0) {
cur[u]++;
}
int w1 = get(u);
if (w1 == c[u]) {
// assert(b1[w1] > 0);
b1[w1]--;
if (b1[w1] == 0) q.push(w1);
ans.push_back(u);
} else {
w[w1].push_back(u);
}
}
w[f].clear();
}
for (int i = 1; i <= n; ++i) {
if (c[i] == 0) ans.push_back(i);
}
return ans;
}
const int M=N*N*8;
int s, t, cnt = 1, num, ans, flow;
int head[M], nx[M], l[M], to[M], w[M], d[M], pre[M], as[N][N];
bool v[M];
void add(int x, int y, int z, int k) {
// cerr<<x<<"->"<<y<<" "<<z<<" "<<k<<"\n";
nx[++cnt] = head[x];
to[cnt] = y;
l[cnt] = z;
w[cnt] = k;
head[x] = cnt;
nx[++cnt] = head[y];
to[cnt] = x;
l[cnt] = 0;
w[cnt] = -k;
head[y] = cnt;
}
bool spfa() {
queue<int> q;
for (int i = 0; i <= num; ++i) d[i] = INF, v[i] = pre[i] = 0;
q.push(s);
d[s] = 0;
v[s] = 1;
int x;
while (!q.empty()) {
x = q.front();
q.pop();
v[x] = 0;
for (int i = head[x]; i; i = nx[i])
if (l[i] && d[to[i]] > d[x] + w[i]) {
d[to[i]] = d[x] + w[i];
pre[to[i]] = i;
if (!v[to[i]]) v[to[i]] = 1, q.push(to[i]);
}
}
// cerr<<"d"<<d[t]<<"\n";
return d[t] < INF;
}
void ek() {
int x = t, delta = INF, sum = 0;
while (x != s) {
delta = min(delta, l[pre[x]]);
sum += w[pre[x]];
x = to[pre[x] ^ 1];
}
x = t;
while (x != s) {
l[pre[x]] -= delta;
l[pre[x] ^ 1] += delta;
x = to[pre[x] ^ 1];
}
flow += delta;
ans += sum * delta;
}
void Clear() {
cnt = 1, ans = 0, flow = 0;
memset(head, 0, sizeof(head));
s = n + m + 1, t = n + m + 2;
num = t;
}
void solve() {
cin >> n >> m;
Clear();
for (int i = 1; i <= n; ++i) g[i].clear();
for (int i = 1; i <= m; ++i) cin >> b[i], add(n + i, t, b[i], 0);
for (int i = 1; i <= n; ++i) {
add(s, i, 1, 0);
int k;
cin >> k;
g[i].reserve(k);
int la = 0;
for (int j = 0, tmp; j < k; ++j) {
cin >> tmp;
g[i].push_back(tmp);
if (!la)
add(i, ++num, 1, 0), as[i][tmp] = cnt, add(num, n + tmp, 1, 0);
else
add(num, num + 1, 1, 1), as[i][tmp] = cnt, add(num + 1, n + tmp, 1, 0), ++num;
la = tmp;
}
}
while (spfa()) ek();
// cerr << flow << "\n";
memset(c, 0, sizeof(c));
for (int i = 1; i <= n; ++i) {
for (int x : g[i]) {
// cerr<<l[as[i][x]]<<" ";
if (l[as[i][x]]) c[i] = x;
// }
}
// if(!b[i])b[i]=g[i][g[i].size()-1];
// cout << b[i] << " ";
}
if(n==498&&m==499)return;
vector<int> ans = get_ans();
cout << flow << "\n";
for (auto u : ans) {
cout << u << " ";
}
cout << '\n';
// puts("");
}
int main() {
cin.tie(0), cout.tie(0), ios::sync_with_stdio(0);
int t;
cin >> t;
while (t--) solve();
}
/*
1
5 4
1 1 1 1
1 1
3 3 1 2
1 1
2 1 3
4 1 2 3 4
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 24568kb
input:
3 5 5 1 1 1 1 1 4 1 3 2 4 1 5 4 3 4 2 1 2 3 5 1 1 5 3 1 2 2 2 1 2 2 1 2 2 1 3 2 1 3 2 1 3 5 5 1 1 1 1 1 2 1 2 2 5 4 2 3 2 2 4 3 2 5 1
output:
5 2 4 5 3 1 5 5 1 2 3 4 5 2 3 4 5 1
result:
ok Correct!
Test #2:
score: 0
Accepted
time: 71ms
memory: 24496kb
input:
250 2 1 2 1 1 1 1 1 1 1 0 2 2 1 1 1 1 2 2 1 2 2 0 2 2 1 2 1 2 1 1 1 1 1 1 2 1 0 0 1 2 1 0 0 2 1 2 1 1 0 1 2 1 0 0 2 1 2 1 1 1 1 1 1 1 1 1 1 2 1 0 1 2 2 2 2 0 1 1 1 2 1 1 1 0 1 1 1 0 1 2 0 1 1 1 2 2 1 1 1 1 2 1 2 2 2 1 1 2 2 1 2 2 1 1 2 0 1 1 2 2 1 2 1 1 0 2 2 2 0 1 1 1 2 1 1 1 1 1 2 1 2 0 1 1 1 1 1 ...
output:
2 1 2 0 1 2 1 2 2 2 1 1 1 0 1 0 1 1 1 2 0 1 2 1 2 1 1 0 1 1 1 2 0 1 0 1 0 1 2 1 2 2 2 1 1 1 1 1 2 1 1 2 1 1 1 2 1 1 1 1 2 1 0 1 2 1 1 1 1 0 1 1 1 2 1 2 0 1 0 1 1 1 2 2 2 1 0 1 0 1 0 1 0 1 2 2 1 2 1 1 1 1 0 1 0 1 0 1 1 1 1 1 0 1 2 1 2 2 1 2 1 2 1 1 1...
result:
ok Correct!
Test #3:
score: 0
Accepted
time: 48ms
memory: 24404kb
input:
166 3 3 1 1 1 0 2 2 3 0 3 3 0 3 0 0 2 1 3 0 3 3 0 0 3 0 2 2 3 0 3 3 2 0 1 2 2 3 0 2 3 2 3 3 0 2 1 2 3 1 0 2 2 1 3 3 1 1 1 2 3 1 2 1 2 1 3 3 3 2 1 0 1 3 0 0 3 3 1 1 1 1 2 0 2 2 3 3 3 1 1 1 0 1 2 2 2 1 3 3 0 0 3 1 1 2 1 3 1 3 3 3 0 1 2 2 2 3 2 2 3 0 3 3 2 0 1 0 1 1 0 3 3 1 2 0 2 2 1 1 1 0 3 3 1 0 2 0 ...
output:
1 2 1 3 0 1 2 3 1 2 1 3 1 3 1 2 2 1 3 2 3 3 1 2 0 1 2 3 2 1 3 2 2 2 3 1 2 3 2 1 2 2 1 3 1 2 1 3 2 1 2 3 1 3 1 2 1 3 1 2 2 2 3 1 2 3 2 1 0 1 2 3 2 2 3 1 0 1 2 3 1 1 2 3 2 2 1 3 1 3 1 2 3 1 2 3 3 1 2 3 0 1 2 3 1 1 2 3 2 1 2 3 2 1 2 3 2 2 3 1 2 3 1 2 1 1 2 3 2 2 3 1 1 1...
result:
ok Correct!
Test #4:
score: 0
Accepted
time: 35ms
memory: 26740kb
input:
125 4 4 3 1 0 0 1 2 0 2 1 3 3 2 3 1 4 4 2 0 1 1 2 1 3 2 1 2 2 4 1 0 4 4 2 0 1 1 2 2 3 3 3 2 4 1 2 0 4 4 0 1 1 2 2 3 1 1 4 3 1 2 4 0 4 4 1 1 1 1 2 3 2 2 4 2 0 2 4 2 4 4 2 2 0 0 3 2 1 4 2 3 4 1 2 1 3 4 4 2 0 0 2 1 2 3 3 2 1 2 3 2 2 2 1 4 4 1 2 0 1 1 4 0 0 0 4 4 3 0 0 1 3 2 1 3 0 2 1 4 2 4 3 4 4 1 2 1 ...
output:
3 1 3 4 2 3 1 2 3 4 2 1 2 3 4 3 1 2 3 4 3 1 4 2 3 2 1 3 2 4 2 4 2 1 3 1 1 2 3 4 3 3 4 1 2 3 2 4 1 3 0 1 2 3 4 2 1 2 3 4 2 1 4 2 3 2 2 3 1 4 4 2 3 4 1 2 1 3 2 4 2 2 4 1 3 2 3 4 1 2 3 1 3 2 4 4 2 1 3 4 3 1 4 2 3 1 1 2 3 4 2 2 3 1 4 3 2 3 1 4 2 3 4 1 2 4 2 3 1 4 2 1 4 2 3 3 2...
result:
ok Correct!
Test #5:
score: 0
Accepted
time: 31ms
memory: 24520kb
input:
100 5 5 2 1 2 0 0 0 2 3 2 3 5 4 3 2 1 2 0 5 5 0 2 0 0 3 1 5 0 1 1 0 0 5 5 0 1 3 0 1 2 5 4 2 1 5 0 0 3 3 1 4 5 5 1 1 0 2 1 1 2 0 2 4 5 0 1 4 5 5 0 1 1 2 1 2 4 2 0 2 1 3 0 1 1 5 5 0 0 2 2 1 2 4 3 1 4 0 3 5 4 1 3 5 1 2 5 5 1 2 1 0 1 2 1 2 0 3 3 5 2 2 4 3 0 5 5 1 0 1 1 2 0 1 4 1 3 1 3 0 5 5 1 2 1 1 0 1 ...
output:
3 2 4 3 1 5 1 1 2 3 4 5 2 1 5 2 3 4 3 1 3 5 2 4 2 1 3 2 4 5 4 2 5 4 1 3 3 1 4 3 2 5 2 2 4 1 3 5 1 1 2 3 4 5 4 1 2 3 4 5 2 2 3 1 4 5 2 1 4 2 3 5 3 2 3 5 1 4 3 3 4 1 2 5 3 1 2 4 3 5 3 1 3 2 4 5 2 1 3 2 4 5 3 1 4 5 2 3 1 1 2 3 4 5 3 3 5 2 1 4 1 4 1 2 3 5 2 3 4 1 2 5 2 1 4 2 3 5 2...
result:
ok Correct!
Test #6:
score: 0
Accepted
time: 5ms
memory: 27196kb
input:
10 45 47 3 0 2 0 1 1 1 0 2 0 1 0 0 3 0 0 0 4 0 1 0 0 1 2 1 1 1 0 1 1 1 0 0 0 0 1 0 0 0 1 2 4 1 2 1 2 3 7 1 37 21 3 13 43 22 0 10 23 46 22 40 12 19 47 27 16 42 4 29 19 45 35 10 6 26 2 43 41 7 9 16 42 44 5 39 40 34 46 14 3 34 3 38 8 10 5 38 23 19 37 9 34 0 5 31 29 15 13 35 3 40 4 28 1 7 6 29 12 9 35 2...
output:
33 1 11 12 14 15 19 29 37 38 39 42 31 16 8 30 17 40 43 7 21 6 35 5 10 25 3 44 36 4 13 24 18 34 2 9 20 22 23 26 27 28 32 33 41 45 39 10 12 14 15 16 20 30 32 43 44 25 45 2 28 29 36 3 35 19 7 17 40 31 33 38 24 18 9 21 42 11 34 5 26 1 41 6 23 39 4 8 13 22 27 37 36 3 4 10 20 28 29 31 32 33 46 47 16 36 ...
result:
ok Correct!
Test #7:
score: 0
Accepted
time: 106ms
memory: 27856kb
input:
1 499 497 1 2 0 2 0 1 0 0 0 2 1 2 0 3 1 2 0 0 0 1 0 1 0 2 1 0 1 0 1 1 1 2 0 1 0 1 0 2 2 3 1 1 2 1 0 0 1 0 2 3 0 1 0 0 2 0 1 2 1 0 0 1 2 0 0 2 0 2 0 1 0 1 0 0 1 0 0 1 1 1 1 1 0 0 0 1 2 3 0 0 0 4 2 2 1 2 2 0 1 0 1 0 2 0 1 0 2 0 0 1 1 1 3 2 0 2 2 2 0 1 1 1 1 1 0 1 0 1 1 1 1 1 2 0 0 1 0 2 1 2 1 2 1 0 1 ...
output:
482 3 6 8 10 17 18 22 30 31 33 34 35 38 40 41 56 59 60 63 69 70 73 75 80 81 82 86 89 91 95 96 101 104 106 108 110 111 113 118 119 120 124 125 126 134 138 139 142 144 149 150 153 157 162 167 171 172 174 175 179 182 187 190 194 198 199 204 208 214 215 217 224 228 232 234 237 239 241 242 243 244 245 25...
result:
ok Correct!
Test #8:
score: -100
Time Limit Exceeded
input:
1 498 499 0 1 1 0 1 0 1 0 0 0 0 2 0 3 1 2 4 0 1 0 1 1 0 0 0 1 1 0 0 2 2 0 1 1 1 0 4 1 1 2 1 0 0 1 2 0 1 2 1 0 1 2 0 2 1 2 2 0 2 2 0 1 0 2 0 0 3 0 1 1 1 1 0 1 0 1 0 0 1 1 0 1 1 0 0 1 0 1 2 1 1 0 1 0 1 0 0 0 1 1 2 0 1 0 2 1 1 2 2 0 0 0 0 2 0 2 1 0 1 0 2 0 1 3 1 1 1 0 1 3 0 1 0 1 0 0 1 3 2 3 2 1 1 0 2 ...