QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#605058#8759. 小班课UESTC_NLNSTL 106ms27856kbC++144.3kb2024-10-02 15:16:142024-10-02 15:16:15

Judging History

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

  • [2024-10-02 15:16:15]
  • 评测
  • 测评结果:TL
  • 用时:106ms
  • 内存:27856kb
  • [2024-10-02 15:16:14]
  • 提交

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

output:


result: