QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#645251#7898. I Just Want... One More...lamkappaWA 12ms3636kbC++204.3kb2024-10-16 17:23:222024-10-16 17:23:24

Judging History

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

  • [2024-10-16 17:23:24]
  • 评测
  • 测评结果:WA
  • 用时:12ms
  • 内存:3636kb
  • [2024-10-16 17:23:22]
  • 提交

answer

#include<bits/stdc++.h>

constexpr int inf = 1E9;
template<class T>
struct MaxFlow {
    struct _Edge {
        int to;
        T cap;
        _Edge(int to, T cap) : to(to), cap(cap) {}
    };

    int n;
    std::vector<_Edge> e;
    std::vector<std::vector<int>> g;
    std::vector<int> cur, h, gap;

    MaxFlow() {}
    MaxFlow(int n) {
        init(n);
    }

    void init(int n) {
        this->n = n;
        e.clear();
        g.assign(n, {});
        cur.resize(n);
        h.resize(n);
        gap.resize(n+2);
    }

    bool bfs(int s, int t) {
        h.assign(n, -1);
        gap.assign(n+2, 0);
        std::queue<int> que;
        h[s] = 0;
        que.push(s);
        while (!que.empty()) {
            const int u = que.front(); que.pop();
            gap[h[u]]++;
            for (int i : g[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;
    }

    void addEdge(int u, int v, T c) {
        g[u].push_back(e.size());
        e.emplace_back(v, c);
        g[v].push_back(e.size());
        e.emplace_back(u, 0);
    }
    T flow(int s, int t) {
        bfs(s, t);
        T ans = 0;
        auto dfs = [&](auto&&dfs, int u, T f) -> T {
            if (u == t || f == T{}) {
                return f;
            }
            auto r = f;
            for (int &i = cur[u]; i < int(g[u].size()); ++i) {
                const int j = g[u][i];
                auto [v, c] = e[j];
                if (c > 0 && h[v] == h[u] + 1) {
                    auto a = dfs(dfs, v, std::min(r, c));
                    e[j].cap -= a;
                    e[j ^ 1].cap += a;
                    r -= a;
                    if (r <= 0) {
                        return f;
                    }
                }
            }
            if(--gap[h[u]] == 0){
                h[s] = n + 1;
            }
            ++gap[++h[u]];
            cur[u] = h[u];
            return f - r;
        };
        while (h[s] < n) {
            cur.assign(n, 0);
            ans += dfs(dfs, s, std::numeric_limits<T>::max());
        }
        return ans;
    }

    std::vector<bool> minCut() {
        std::vector<bool> c(n);
        for (int i = 0; i < n; i++) {
            c[i] = (h[i] != -1);
        }
        return c;
    }

    long long getAns(int N, int s, int t) {
        std::vector<bool> vis_t(N * 2 + 3, false);
        std::queue<int> q;
        int s1 = 0, s2 = 0;
        q.push(s);
        vis_t[s] = true;
        while(!q.empty()) {
            int x = q.front();
            if(x != s && x != t && x <= N) {
                s1++;
            }
            q.pop();
            for(auto i : g[x]) {
                auto [to, cap] = e[i];
                if(cap == 0 || vis_t[to]) {
                    continue;
                }
                vis_t[to] = true;
                q.push(to);
            }
        }
        q = {};
        std::fill(vis_t.begin(), vis_t.end(), false);
        q.push(t);
        vis_t[t] = true;
        while(!q.empty()) {
            int x = q.front();
            if(x != s && x != t && x > N) {
                s2++;
            }
            q.pop();
            for(auto i : g[x]) {
                auto [to, cap] = e[i];
                if(cap == 1 || vis_t[to]) {
                    continue;
                }
                vis_t[to] = true;
                q.push(to);
            }
        }
        return 1ll * s1 * s2;
    }
};
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    std::cin >> T;
    while(T--) {
        int n, m;
        std::cin >> n >> m;
        int s = n * 2 + 1, t = n * 2 + 2;
        MaxFlow<int> g(n * 2 + 3);
        for(int x, y, i = 1; i <= m; i++) {
            std::cin >> x >> y;
            g.addEdge(x, n + y, 1);
        }
        for(int i = 1; i <= n; i++) {
            g.addEdge(s, i, 1);
            g.addEdge(i + n, t, 1);
        }
        g.flow(s, t);
        std::cout << g.getAns(n, s, t) << '\n';
    }

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

3
4 3
1 2
3 2
4 3
3 3
1 3
2 2
3 1
3 2
1 2
1 2

output:

6
0
4

result:

ok 3 number(s): "6 0 4"

Test #2:

score: -100
Wrong Answer
time: 12ms
memory: 3636kb

input:

10000
5 4
2 2
3 1
5 3
1 1
1 1
1 1
1 1
1 1
2 2
2 2
1 2
1 1
1 1
1 1
1 1
1 1
1 1
2 3
2 1
1 2
1 1
5 5
1 4
2 2
3 1
1 3
1 2
2 4
2 2
2 1
1 2
1 1
5 1
5 3
3 1
2 2
1 1
1 1
3 2
3 1
2 1
5 2
1 2
2 3
5 3
1 5
4 2
1 2
4 1
1 1
2 3
1 1
2 2
2 1
4 1
1 4
3 1
1 1
1 1
1 1
2 1
2 2
3 3
1 3
2 3
2 2
3 3
1 3
3 3
1 2
3 3
2 2
1 ...

output:

25
0
0
2
0
0
0
0
6
0
16
4
0
6
9
9
9
0
9
4
0
1
1
9
0
4
16
12
9
2
16
0
2
2
20
1
9
0
0
0
16
4
4
16
4
9
0
9
0
2
3
0
9
4
9
16
20
0
0
1
12
0
9
2
0
0
1
0
0
16
2
4
0
12
1
0
0
2
1
2
2
9
0
9
1
6
0
0
0
0
9
16
2
0
1
2
0
12
2
4
0
12
1
1
9
16
6
9
9
12
3
16
15
16
9
4
9
0
1
16
9
9
1
9
16
9
12
4
9
2
0
16
0
6
0
3
0
0...

result:

wrong answer 1st numbers differ - expected: '6', found: '25'