結果

問題 No.3201 Corporate Synergy
ユーザー k1suxu
提出日時 2025-07-11 22:20:00
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 4,738 bytes
コンパイル時間 3,713 ms
コンパイル使用メモリ 290,716 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-07-11 22:20:04
合計ジャッジ時間 4,065 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

// #pragma GCC target("avx")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")

#include <bits/stdc++.h>
using namespace std;

using std::cin;
using std::cout;

#define rep(i,n) for(int i = 0; i < (int)n; i++)
#define FOR(n) for(int i = 0; i < (int)n; i++)
#define repi(i,a,b) for(int i = (int)a; i < (int)b; i++)
#define all(x) x.begin(),x.end()
//#define mp make_pair
#define vi vector<int>
#define vvi vector<vi>
#define vvvi vector<vvi>
#define vvvvi vector<vvvi>
#define pii pair<int,int>
#define vpii vector<pair<int,int>>

template<typename T>
bool chmax(T &a, const T b) {if(a<b) {a=b; return true;} else {return false;}}
template<typename T>
bool chmin(T &a, const T b) {if(a>b) {a=b; return true;} else {return false;}}

using ll = long long;
using ld = long double;
using ull = unsigned long long;

const ll INF = numeric_limits<long long>::max() / 2;
const ld pi = 3.1415926535897932384626433832795028;
const ll mod = 998244353;
int dx[] = {1, 0, -1, 0, -1, -1, 1, 1};
int dy[] = {0, 1, 0, -1, -1, 1, -1, 1};

#define int long long

template<typename T>
struct Edge {
    int to;
    T cap;
    int rev;

    Edge() = default;
    Edge(int to, T cap, int rev) : to(to), cap(cap), rev(rev) {}
};

template<typename T>
struct Dinic {
    const T e = numeric_limits<T>::max();
    const T zero = 0;

    int n;
    vector<vector<Edge<T>>> g;
    vector<int> level, iter;
    vector<pair<int, int>> Edge_ID;

    Dinic(int n) : n(n), g(n) {}

    int add_edge(int from, int to, T cap) {
        Edge<T> f(to, cap, g[to].size());
        Edge<T> t(from, zero, g[from].size());

        int Now_Edge_ID = (int)Edge_ID.size();
        Edge_ID.emplace_back(from, (int)g[from].size());

        g[from].push_back(f);
        g[to].push_back(t);
        
        return Now_Edge_ID;
    }

    bool bfs(int s, int t) {
        level.assign(n, -1);
        level[s] = 0;
        queue<int> que;
        que.push(s);

        while(que.size()) {
            int v = que.front();
            que.pop();

            for(auto e : g[v]) {
                if(e.cap > zero && level[e.to] < 0) {
                    level[e.to] = level[v] + 1;
                    que.push(e.to);
                }
            }
        }

        return level[t] != -1;
    }

    T dfs(int v, int t, T f) {
        if(v == t) return f;

        for(int& i = iter[v]; i < g[v].size(); i++) {
            Edge<T>& e = g[v][i];

            if(e.cap > zero && level[v] < level[e.to]) {
                T d = dfs(e.to, t, min(f, e.cap));

                if(d > zero) {
                    e.cap -= d;
                    g[e.to][e.rev].cap += d;
                    return d;
                }
            }
        }

        return zero;
    }

    T flow(int s, int t) {
        T f = zero;

        while(bfs(s, t)) {
            iter.assign(n, 0);
            T d;

            while((d = dfs(s, t, e)) > zero) f += d;
        }

        return f;
    }

    //sから見たときのmin_cutにおいてs側に所属するかどうかを全頂点に対して返すO(N)
    vector<bool> min_cut(int s) {
        vector<bool> visited(n, false);
        queue<int> que;
        visited[s] = true;
        que.push(s);
        while(!que.empty()) {
            int v = que.front();
            que.pop();
            for(auto e : g[v]) if(e.cap != 0 && !visited[e.to]) {
                visited[e.to] = true;
                que.push(e.to);
            }
        }
        return visited;
    }
};

//bipartite matching 
//TODO: make faster Dinic using dynamic tree structure and current edge structure
//燃やす埋めるref: https://qiita.com/ningenMe/items/69ed7ce43c9cd0a2de38

//add_edgeにintの返り値を持たせて、id = add_edge(), get_edge(id) = 元の辺 みたいにしたい。
//min_cut : s側ならばtrueのvector<bool>を返す。
//cut[from] = true && cut[to] = false --- from->toの辺が最小カットで切られる辺の一つ

void solve() {
    int n;
    cin >> n;
    Dinic<int> dinic(n + 2);
    int src = n, dst = n + 1;
    vi p(n);
    int base = 0;
    FOR(n) {
        cin >> p[i];
        if (p[i] > 0) base += p[i], dinic.add_edge(i, dst, p[i]);
        else dinic.add_edge(src, i, -p[i]);
    }
    int m;
    cin >> m;
    FOR(m) {
        int u, v;
        cin >> u >> v;
        u--;
        v--;
        dinic.add_edge(u, v, INF);
    }
    int k;
    cin >> k;
    FOR(k) {
        int a, b, s;
        cin >> a >> b >> s;
        a--;
        b--;
        base += s;
        dinic.add_edge(b, a, s);
        dinic.add_edge(a, dst, s);
    }
    cout << base - dinic.flow(src, dst) << endl;
}

signed main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    solve();
    return 0;
}
0