QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#605028 | #8759. 小班课 | UESTC_NLNS | TL | 102ms | 21660kb | C++14 | 4.3kb | 2024-10-02 15:06:56 | 2024-10-02 15:06:56 |
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 = 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] << " ";
}
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
*/
詳細信息
Test #1:
score: 100
Accepted
time: 3ms
memory: 17552kb
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: 7ms
memory: 18624kb
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: 8ms
memory: 17536kb
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: 3ms
memory: 18692kb
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: 18412kb
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: 3ms
memory: 17540kb
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: 102ms
memory: 21660kb
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 ...