結果

問題 No.1984 [Cherry 4th Tune *] Dilemma
ユーザー 👑 Nachia
提出日時 2024-04-26 02:06:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 1,689 bytes
コンパイル時間 1,407 ms
コンパイル使用メモリ 97,992 KB
最終ジャッジ日時 2025-02-21 09:04:17
ジャッジサーバーID
(参考情報)
judge2 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 68
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <atcoder/modint>

#include <atcoder/maxflow>

using namespace std;
using i64 = long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)


const i64 INF = 1001001001001001001;
using modint = atcoder::static_modint<1000000007>;


int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int N; cin >> N;
    int M; cin >> M;
    int K; cin >> K;
    int P; cin >> P;
    int nodeIdxSrc = 0;
    int nodeIdxN = nodeIdxSrc + 1;
    int nodeIdxM = nodeIdxN + N;
    int nodeIdxK = nodeIdxM + M;
    int nodeIdxSnk = nodeIdxK + K;

    atcoder::mf_graph<i64> G(nodeIdxSnk + 1);
    i64 offset = 0;

    rep(i,N){ int a; cin >> a; G.add_edge(nodeIdxSrc, nodeIdxN+i, a); offset += a; }
    rep(i,M){ int a; cin >> a; G.add_edge(nodeIdxM+i, nodeIdxSnk, a); offset += a; }
    rep(i,K){ int a; cin >> a; G.add_edge(nodeIdxK+i, nodeIdxSnk, a); }
    rep(i,N){
        int L; cin >> L;
        rep(j,L){ int k; cin >> k; k--; G.add_edge(nodeIdxN+i, nodeIdxK+k, INF); }
    }
    rep(i,P){
        int n; cin >> n; n--;
        int m; cin >> m; m--;
        G.add_edge(nodeIdxN+n, nodeIdxM+m, INF);
    }

    i64 ans = offset - G.flow(nodeIdxSrc, nodeIdxSnk);
    auto cut = G.min_cut(nodeIdxSrc);

    cout << ans << '\n';

    vector<string> ansarr;
    rep(i,K) if(cut[nodeIdxK+i]) ansarr.push_back("Preparation " + to_string(i+1));
    rep(i,N) if(cut[nodeIdxN+i]) ansarr.push_back("Goal " + to_string(i+1));
    rep(i,M) if(!cut[nodeIdxM+i]) ansarr.push_back("Action " + to_string(i+1));

    cout << ansarr.size() << '\n';
    for(auto& a : ansarr) cout << a << '\n';
    return 0;
}
0