結果

問題 No.2320 Game World for PvP
コンテスト
ユーザー Nachia
提出日時 2023-06-14 12:26:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,234 bytes
コンパイル時間 1,216 ms
コンパイル使用メモリ 94,048 KB
最終ジャッジ日時 2025-02-14 02:34:10
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <atcoder/modint>
#include <atcoder/maxflow>
using namespace std;
using i32 = int;
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
const i64 INF = 1001001001001001001;

using Modint = atcoder::static_modint<998244353>;

int main(){
    int N, S, T; cin >> N >> S >> T;
    vector<int> col(N, -1);
    rep(i,S){
        int a; cin >> a;
        col[a-1] = 0;
    }
    rep(i,T){
        int a; cin >> a;
        col[a-1] = 1;
    }
    vector<vector<i64>> C(N, vector<i64>(N));
    rep(i,N) rep(j,N) cin >> C[i][j];
    atcoder::mf_graph<i64> mf(N+2);
    i64 off = 0;
    rep(i,N) rep(j,N) if(i < j){
        mf.add_edge(i, j, C[i][j]);
        mf.add_edge(j, i, C[i][j]);
        off += C[i][j];
    }
    rep(i,N){
        if(col[i] == 0) mf.add_edge(N, i, INF);
        if(col[i] == 1) mf.add_edge(i, N+1, INF);
    }
    i64 flow = mf.flow(N, N+1);
    i64 ans = off - flow;
    cout << ans << endl;
    return 0;
}



struct ios_do_not_sync{
    ios_do_not_sync(){
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0