QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#605232#8759. 小班课UESTC_NLNSTL 548ms21360kbC++145.3kb2024-10-02 16:17:512024-10-02 16:17:52

Judging History

你现在查看的是最新测评结果

  • [2024-10-02 16:17:52]
  • 评测
  • 测评结果:TL
  • 用时:548ms
  • 内存:21360kb
  • [2024-10-02 16:17:51]
  • 提交

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 ...

output:


result: