QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#605109 | #8759. 小班课 | UESTC_NLNS | TL | 66ms | 21376kb | C++14 | 4.4kb | 2024-10-02 15:31:28 | 2024-10-02 15:31:28 |
Judging History
answer
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int N = 1505;
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;
}
int s, t, cnt = 1, num, ans, flow;
int head[600 * 600], nx[N * N], l[N * N], to[N * N], w[N * N], d[600 * 600], pre[600 * 600], as[N][N];
bool v[600 * 600];
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 = 1, tmp; j <= k; ++j) {
cin >> tmp;
g[i].push_back(tmp);
// if (!la)
add(i, n + tmp, 1, j);
as[i][tmp] = cnt;
// 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] << " ";
}
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: 15388kb
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: 9ms
memory: 19368kb
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: 10ms
memory: 16928kb
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: 4ms
memory: 18912kb
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: 3ms
memory: 15580kb
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: 6ms
memory: 15524kb
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: 66ms
memory: 21376kb
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 ...