結果

問題 No.3201 Corporate Synergy
ユーザー YY-otter
提出日時 2025-07-03 08:56:09
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,403 bytes
コンパイル時間 1,382 ms
コンパイル使用メモリ 124,364 KB
実行使用メモリ 814,084 KB
最終ジャッジ日時 2025-07-06 10:24:50
合計ジャッジ時間 4,304 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 2
other RE * 4 MLE * 1 -- * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <numeric>
#include <tuple>

// AtCoder Libraryから最大流ライブラリをインクルード
#include <atcoder/maxflow>

using ll = long long;
const ll INF = 1e18; 

int main() {
    // 高速な入出力
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(NULL);

    // 1. 入力読み込み
    int N, M, K;
    std::cin >> N >> M >> K;

    std::vector<ll> P(N + 1);
    for (int i = 1; i <= N; ++i) {
        std::cin >> P[i];
    }
    std::vector<std::pair<int, int>> dependencies(M);
    for (int i = 0; i < M; ++i) {
        std::cin >> dependencies[i].first >> dependencies[i].second;
    }
    std::vector<std::tuple<int, int, ll>> partnerships(K);
    for (int i = 0; i < K; ++i) {
        std::cin >> std::get<0>(partnerships[i]) >> std::get<1>(partnerships[i]) >> std::get<2>(partnerships[i]);
    }

    // 2. グラフ構築（前回と全く同じ）
    int S = 0, T = N + K + 1;
    atcoder::mf_graph<ll> graph(T + 1);

    for (int i = 1; i <= N; ++i) {
        if (P[i] > 0) graph.add_edge(S, i, P[i]);
        else graph.add_edge(i, T, -P[i]);
    }
    for (const auto& dep : dependencies) {
        graph.add_edge(dep.second, dep.first, INF);
    }
    for (int i = 0; i < K; ++i) {
        auto [u, v, s] = partnerships[i];
        int p_node = N + 1 + i;
        if (s > 0) {
            graph.add_edge(S, p_node, s);
            graph.add_edge(p_node, u, INF);
            graph.add_edge(p_node, v, INF);
        } else {
            graph.add_edge(p_node, T, -s);
            graph.add_edge(u, p_node, INF);
            graph.add_edge(v, p_node, INF);
        }
    }

    // 3. 最小カットを計算
    graph.flow(S, T);

    // 4. 最適な事業選択を特定し、定義通りに総利益を直接計算する
    ll max_profit = 0;
    std::vector<bool> s_side = graph.min_cut(S); // S側の頂点集合を取得

    // S側にある都市のP_iを合計
    for (int i = 1; i <= N; ++i) {
        if (s_side[i]) {
            max_profit += P[i];
        }
    }
    // S側にある提携のS_jを合計
    for (int i = 0; i < K; ++i) {
        auto [u, v, s] = partnerships[i];
        // 提携の条件（uとvの両方がS側）を満たすかチェック
        if (s_side[u] && s_side[v]) {
            max_profit += s;
        }
    }

    std::cout << max_profit << std::endl;

    return 0;
}
0