QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#378298#8566. Can We Still Qualify For Semifinals?ucup-team2112#AC ✓2ms3852kbC++205.5kb2024-04-06 10:58:142024-04-06 10:58:15

Judging History

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

  • [2024-04-06 10:58:15]
  • 评测
  • 测评结果:AC
  • 用时:2ms
  • 内存:3852kb
  • [2024-04-06 10:58:14]
  • 提交

answer

#include <bits/stdc++.h>

template<class T>
struct Dinic {
    const int n;
    struct edge {
        int to;
        T cap;
        edge(int to, T cap) : to(to), cap(cap) {}
    };
    std::vector<edge> e;
    std::vector<std::vector<int> > adj;
    std::vector<int> cur, h;
    Dinic(int n) : n(n), adj(n) {}
    
    bool bfs(int s, int t) {
        h.assign(n, -1);
        std::queue<int> que;
        h[s] = 0;
        que.push(s);
        while (!que.empty()) {
            const int u = que.front();
            que.pop();
            for (int i : adj[u]) {
                auto [v, c] = e[i];
                if (c > 0 && h[v] == -1) {
                    h[v] = h[u] + 1;
                    if (v == t) {
                        return true;
                    }
                    que.push(v);
                }
            }
        }
        return false;
    }
    
    T dfs(int u, int t, T f) {
        if (u == t) {
            return f;
        }
        auto r = f;
        for (int &i = cur[u]; i < int(adj[u].size()); ++i) {
            const int j = adj[u][i];
            auto [v, c] = e[j];
            if (c > 0 && h[v] == h[u] + 1) {
                auto a = dfs(v, t, std::min(r, c));
                e[j].cap -= a;
                e[j ^ 1].cap += a;
                r -= a;
                if (r == 0) {
                    return f;
                }
            }
        }
        return f - r;
    }
    void addedge(int u, int v, T c) {
        adj[u].push_back(e.size());
        e.emplace_back(v, c);
        adj[v].push_back(e.size());
        e.emplace_back(u, 0);
    }
    T maxflow(int s, int t) {
        T ans = 0;
        while (bfs(s, t)) {
            cur.assign(n, 0);
            ans += dfs(s, t, std::numeric_limits<T>::max());
        }
        return ans;
    }
};

void solve(){
    int n;
    std::string s;
    std::cin >> n >> s;
    std::vector<std::pair<int, int> > match;
    {
        std::vector<int> team(10);
        std::iota(team.begin(), team.end(), 0);
        for (int i = 0; i < 9; i += 1) {
            for (int l = 0, r = 9; l < r; l += 1, r -= 1) {
               match.emplace_back(team[l], team[r]);
            //    std::cerr << team[l] + 1 << " " << team[r] + 1 << "\n";
            }
            std::reverse(team.begin() + 1, team.end());
            std::rotate(team.begin() + 1, team.begin() + 2, team.end());
            std::reverse(team.begin() + 1, team.end());
            // for (auto x : team) {
            //     std::cerr << x << " ";
            // }
            // std::cerr << "\n";;
        }
    }
    std::reverse(match.begin(), match.end());
    std::vector<int> win(10);
    for (auto c : s) {
        auto [a, b] = match.back();
        std::cerr << a + 1 << " " << b + 1 << "\n";
        match.pop_back();
        if (c == '1') {
            win[a] += 1;
        }
        else {
            win[b] += 1;
        }
    }
    {
        decltype(match) a;
        for (int i = 0; i < match.size(); i += 1) {
            auto [x, y] = match[i];
            if (x == 0 or y == 0) {
                win[0] += 1;
            }
            else {
                a.emplace_back(x, y);
            }
        }
        match = a;
    }
    int v = win[0];
    for (int i = 1; i < 10; i += 1) {
        for (int j = i + 1; j < 10; j += 1) {
            for (int k = j + 1; k < 10; k += 1) {
                decltype(match) new_match;
                for (auto [x, y] : match) {
                    if (x == i or x == j or x == k or y == i or y == j or y == k) {
                        continue;
                    }
                    new_match.emplace_back(x, y);
                }
                bool nice = true;
                for (int x = 1; x < 10; x += 1) {
                    if (x == i or x == j or x == k) {
                        continue;
                    }
                    if (win[x] > v) {
                        nice = false;
                        break;
                    }
                }
                if (not nice) {
                    continue;
                }
                int nl = new_match.size();
                int nr = 6;
                Dinic<int> dinic(nl + nr + 2);
                for (int i = 1; i <= nl; i += 1) {
                    dinic.addedge(0, i, 1);
                }
                std::vector<int> label(10);
                for (int x = 1, c = 0; x < 10; x += 1) {
                    if (x == i or x == j or x == k) {
                        continue;
                    }
                    c += 1;
                    label[x] = c;
                    dinic.addedge(nl + c, nl + nr + 1, v - win[x]);
                }

                for (int i = 0; i < nl; i += 1) {
                    auto [x, y] = new_match[i];
                    x = label[x];
                    y = label[y];
                    dinic.addedge(i + 1, nl + x, 1);
                    dinic.addedge(i + 1, nl + y, 1);
                }
                
                int res = dinic.maxflow(0, nl + nr + 1);
                if (res == nl) {
                    std::cout << "YES\n";
                    return;
                }
            }
        }
    }
    std::cout << "NO\n";
}

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int T;
    std::cin >> T;
    while(T--) {
        solve();
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3800kb

input:

3
3
111
25
1000010101111111111010100
35
01111011110111101111011110111101111

output:

YES
YES
NO

result:

ok 3 token(s): yes count is 2, no count is 1

Test #2:

score: 0
Accepted
time: 0ms
memory: 3604kb

input:

10
16
0110000001010100
17
01111000110110101
15
001100010101111
16
0010101010011100
19
0000000100010110100
16
0011101010011100
18
011110010001100000
18
000110101001100011
20
01100010000100100100
15
001000111001101

output:

YES
YES
YES
YES
YES
YES
YES
YES
YES
YES

result:

ok 10 token(s): yes count is 10, no count is 0

Test #3:

score: 0
Accepted
time: 0ms
memory: 3600kb

input:

10
37
0110000001010100011101001011100110001
39
000100111101101001100101101000000000100
35
00111000100111100101011010111100100
33
010000010001110010110001101110001
30
000100010100000010010110101010
31
0000101000011010101001010000000
44
00001000000111101011010110000101100011000100
42
01111011110001001...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO

result:

ok 10 token(s): yes count is 0, no count is 10

Test #4:

score: 0
Accepted
time: 2ms
memory: 3632kb

input:

10
23
01100000010101000111010
38
01111001100011000101011110101001101001
27
010000000001001001110001001
26
01101001110011101101000110
8
00001000
22
0110100110001110110001
9
000100010
24
000000100101101010100100
6
011000
29
01101010100101000000000000100

output:

YES
NO
NO
NO
YES
YES
YES
YES
YES
NO

result:

ok 10 token(s): yes count is 6, no count is 4

Test #5:

score: 0
Accepted
time: 2ms
memory: 3576kb

input:

10
30
011000000101010001110100101110
29
01001010010011101110010110010
28
0110000000001000101101001001
23
01101001110011101101000
23
01000001000111001011000
24
011110001000010001010000
23
01001011010101001000011
30
000110011001010010000000000010
24
000110111001110011000011
28
000110001000011011110110...

output:

NO
NO
NO
YES
YES
YES
YES
NO
YES
NO

result:

ok 10 token(s): yes count is 5, no count is 5

Test #6:

score: 0
Accepted
time: 2ms
memory: 3596kb

input:

10
21
011000000101010001110
21
000110110101001010010
22
0111101101001100101101
24
000000001000101011000101
21
011010011100111011010
20
00110000010001101010
21
010010111100010000100
24
010100000100011010110010
23
00001010000110101010010
25
0000000000001000001101110

output:

YES
YES
YES
YES
YES
YES
YES
YES
YES
YES

result:

ok 10 token(s): yes count is 10, no count is 0

Test #7:

score: 0
Accepted
time: 2ms
memory: 3836kb

input:

10
26
01100000010101000111010010
26
01101010010100100111011100
26
00110010110100000000010010
27
011100010101110010110101101
30
010100011000001000110101001100
30
011110001000010001010000001001
28
0101100101000010100001101010
26
00101000000000000100000110
28
0110101101010000111000110001
27
00011011110...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO

result:

ok 10 token(s): yes count is 0, no count is 10

Test #8:

score: 0
Accepted
time: 2ms
memory: 3832kb

input:

10
25
0010100010011010111001111
26
01001010100010101010001010
26
01111001110000100111011110
26
10001000100110101110011110
26
10101010100110101110011110
27
110100010101010011010111001
27
101010101001101011100111101
31
1000010001010100110001011011110
37
1000101111000100110000011000000100101
40
1000101...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO

result:

ok 10 token(s): yes count is 0, no count is 10

Test #9:

score: 0
Accepted
time: 2ms
memory: 3852kb

input:

10
26
00001010000000000000000000
26
00000010000000000000000000
26
01101010100010101011011110
26
00011011110111101111011110
27
001100110101011001110111101
27
000110111101111011110111101
28
0110001001000010011101111011
29
01000000010001101000011110111
29
01000000010000101101011110111
30
01000011110111...

output:

YES
YES
YES
YES
YES
YES
YES
YES
YES
NO

result:

ok 10 token(s): yes count is 9, no count is 1

Test #10:

score: 0
Accepted
time: 2ms
memory: 3548kb

input:

10
1
0
2
00
10
0001101110
14
00101010000011
20
00000010010100101010
25
0000000101000100100001111
35
01110011010000101010000010010000100
40
0000100110001110101100001001000110000001
44
01011010110010101110011000010001010011100011
45
010010001001010011110111101011011000000100001

output:

YES
YES
YES
YES
YES
YES
NO
NO
NO
NO

result:

ok 10 token(s): yes count is 6, no count is 4

Extra Test:

score: 0
Extra Test Passed