QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#605232 | #8759. 小班课 | UESTC_NLNS | TL | 548ms | 21360kb | C++14 | 5.3kb | 2024-10-02 16:17:51 | 2024-10-02 16:17:52 |
Judging History
answer
#include <limits>
#include <queue>
#include <vector>
using namespace std;
using i64 = long long;
using ll = long long;
const int maxn = 600 * 600;
const int maxm = 600 * 600 * 2;
const int inf = 0x3f3f3f3f;
struct MCMF {
struct edge {
int to, cap, cost, rev;
// edge() {}
// edge(int to, int _cap, int _cost, int _rev) : to(to), cap(_cap), cost(_cost), rev(_rev) {}
};
int V, H[maxn + 5], dis[maxn + 5], PreV[maxn + 5], PreE[maxn + 5];
vector<edge> G[maxn + 5];
void init(int n) {
V = n;
for (int i = 0; i <= V; ++i) G[i].clear();
}
int add(int from, int to, int cap, int cost) {
int t = G[from].size();
G[from].emplace_back(edge{to, cap, cost, (int)G[to].size()});
G[to].emplace_back(edge{from, 0, -cost, (int)G[from].size() - 1});
return t;
}
int mcmf(int s, int t) {
int res = 0, flow = 0;
fill(H, H + 1 + V, 0);
int f = inf;
while (f) {
priority_queue<pair<int, int>> q;
fill(dis, dis + 1 + V, inf);
dis[s] = 0;
q.push(pair<int, int>(0, s));
while (!q.empty()) {
pair<int, int> now = q.top();
q.pop();
int v = now.second, Size;
if (dis[v] < -now.first) continue;
Size = G[v].size();
for (int i = 0; i < Size; ++i) {
edge e = G[v][i];
if (e.cap > 0 && dis[e.to] > dis[v] + e.cost + H[v] - H[e.to]) {
dis[e.to] = dis[v] + e.cost + H[v] - H[e.to];
PreV[e.to] = v;
PreE[e.to] = i;
q.push(pair<int, int>(-dis[e.to], e.to));
}
}
}
if (dis[t] == inf) break;
for (int i = 0; i <= V; ++i) H[i] += dis[i];
int d = f;
for (int v = t; v != s; v = PreV[v]) d = min(d, G[PreV[v]][PreE[v]].cap);
f -= d;
flow += d;
res += d * H[t];
for (int v = t; v != s; v = PreV[v]) {
edge& e = G[PreV[v]][PreE[v]];
e.cap -= d;
G[v][e.rev].cap += d;
}
}
return flow;
}
};
#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;
int num;
void INIT() {
s = n + m + 1, t = n + m + 2;
num = t;
}
using pii = pair<int, int>;
void solve() {
cin >> n >> m;
INIT();
vector<vector<pii>> as(n + 1, vector<pii>(m + 1));
MCMF mcfg;
mcfg.init(maxn);
for (int i = 1; i <= n; ++i) g[i].clear();
for (int i = 1; i <= m; ++i) cin >> b[i], mcfg.add(n + i, t, b[i], 0);
for (int i = 1; i <= n; ++i) {
mcfg.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)
mcfg.add(i, ++num, 1, 0), as[i][tmp] = {num, mcfg.add(num, n + tmp, 1, 0)};
else
mcfg.add(num, num + 1, 1, 1), as[i][tmp] = {num + 1, mcfg.add(num + 1, n + tmp, 1, 0)}, ++num;
la = tmp;
}
}
memset(c, 0, sizeof(c));
int flow = mcfg.mcmf(s, t);
for (int i = 1; i <= n; ++i) {
// int lst = 0;
for (int x : g[i]) {
if (mcfg.G[as[i][x].first][as[i][x].second].cap == 0) c[i] = x;
// lst = x;
}
}
cout << flow << '\n';
vector<int> ans = get_ans();
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: 9ms
memory: 17712kb
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: 362ms
memory: 17952kb
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: 264ms
memory: 17920kb
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: 234ms
memory: 17800kb
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: 197ms
memory: 17772kb
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: 118ms
memory: 17760kb
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: 548ms
memory: 21360kb
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 10 17 18 22 30 31 33 34 38 40 41 55 56 59 60 63 69 70 73 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 153 157 162 167 171 172 174 179 182 187 190 199 204 208 214 215 217 224 225 226 227 228 231 232 234 237 239 241 242 243 244 245 250 252 253 ...
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 ...