結果

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

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <numeric>
#include <tuple>
#include <atcoder/maxflow>

using ll = long long;
// INFは全てのありうる利益・コストの合計よりも十分に大きい値に設定
const ll INF = 201LL * 1000000000LL + 1001LL * 1000000000LL;

int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(NULL);

    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]);
    }

    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) {
        int u = dep.first;
        int v = dep.second;
        graph.add_edge(v, u, 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);
        }
        // 副作用のロジックは削除
    }

    graph.flow(S, T);

    ll max_profit = 0;
    std::vector<bool> s_side = graph.min_cut(S);

    for (int i = 1; i <= N; ++i) {
        if (s_side[i]) {
            max_profit += P[i];
        }
    }
    for (int i = 0; i < K; ++i) {
        auto [u, v, s] = partnerships[i];
        if (s_side[u] && s_side[v]) {
            max_profit += s;
        }
    }

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

    return 0;
}
0