結果

問題 No.654 Air E869120
ユーザー RiRinbaruRiRinbaru
提出日時 2024-06-28 11:53:02
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 345 ms / 2,000 ms
コード長 4,734 bytes
コンパイル時間 2,364 ms
コンパイル使用メモリ 194,464 KB
実行使用メモリ 207,428 KB
最終ジャッジ日時 2024-06-28 11:53:08
合計ジャッジ時間 5,383 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 3 ms
6,940 KB
testcase_10 AC 8 ms
6,940 KB
testcase_11 AC 5 ms
6,940 KB
testcase_12 AC 6 ms
6,944 KB
testcase_13 AC 5 ms
6,944 KB
testcase_14 AC 5 ms
6,940 KB
testcase_15 AC 6 ms
6,940 KB
testcase_16 AC 54 ms
6,940 KB
testcase_17 AC 47 ms
6,944 KB
testcase_18 AC 36 ms
6,944 KB
testcase_19 AC 62 ms
6,944 KB
testcase_20 AC 22 ms
9,088 KB
testcase_21 AC 23 ms
8,960 KB
testcase_22 AC 15 ms
8,960 KB
testcase_23 AC 20 ms
9,088 KB
testcase_24 AC 27 ms
9,216 KB
testcase_25 AC 15 ms
8,960 KB
testcase_26 AC 17 ms
9,088 KB
testcase_27 AC 20 ms
12,160 KB
testcase_28 AC 23 ms
12,032 KB
testcase_29 AC 20 ms
12,032 KB
testcase_30 AC 321 ms
207,412 KB
testcase_31 AC 318 ms
207,392 KB
testcase_32 AC 345 ms
207,352 KB
testcase_33 AC 316 ms
207,428 KB
testcase_34 AC 325 ms
207,348 KB
testcase_35 AC 2 ms
6,944 KB
testcase_36 AC 2 ms
6,940 KB
testcase_37 AC 2 ms
6,944 KB
testcase_38 AC 2 ms
6,940 KB
testcase_39 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, p, n) for (ll i = p; i < (ll)(n); i++)
#define rep2(i, p, n) for(ll i = p; i >= (ll)(n); i-- )
using namespace std;
using ll = long long;
using ld = long double;
double pi=3.141592653589793;
const long long inf=2*1e9;
const long long linf=4*1e18;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }



// ford_fulkersonのアルゴリズム
// 事前準備: ffを定義
// 入力: ff.add_edge(a, b, c); // from, to, costで辺を生成
// 出力: ff.maxFlow(A, B) // 頂点A -> Bへの最大流

ll INF = 1e9+1;

// 辺の構造体
struct Edge
{
    // rev: toから行ける頂点のうち、toから見てfromが何番目に位置するか
    // G[from].size() == rev
    ll rev, from, to, cap;
};

// フォードファルカーソン法
class FordFulkerson
{
public:
    vector<vector<Edge>> G;
    vector<bool> visited;
    // 頂点数 n の残余グラフを用意
    ll size = 0;
    void init(ll n)
    {
        G.resize(n);
        visited.resize(n);
        size = n;
    }
    /*
        頂点 u -> v について 上限 cost の辺を追加
        コスト0の逆辺も張る
    */
    void add_edge(ll u, ll v, ll cost)
    {
        ll u_vID = G[u].size();                 // 現時点での G[u] の要素数 = uからみたvのindex
        ll v_uID = G[v].size();                 // 現時点での G[v] の要素数 = vからみたuのindex
        G[u].push_back(Edge{v_uID, u, v, cost}); //<u,v>の逆辺<v,u>はG[u][v_uID]
        G[v].push_back(Edge{u_vID, v, u, 0});    // 逆辺は追加時はコスト0!!
    }
    /*
        深さ優先探索 F はスタートした頂点からposに到達する過程での
        "残余グラフの辺の容量" の最小値）
        goalまでの往路は頂点を記録しながらs->tまでに共通して流せる容量
                                     = s->tまでの容量の最小値を取得
        復路はs->tまでの容量の最小値を使って残余ネットワークのコストを更新
        返り値: 流したフローの量
    */
    ll dfs(ll pos, ll goal, ll F)
    {
        if (pos == goal)
            return F;        // ゴールに到着したら流す
        visited[pos] = true; // 訪れた頂点を記録

        // G[pos]に隣接する頂点を探索
        for (auto &e : G[pos])
        {
            // 容量0の辺や訪問済みの頂点は無視
            if (e.cap == 0 or visited[e.to])
                continue;
            // 再帰で目的地までのパスを探す
            ll flow = dfs(e.to, goal, min(F, e.cap));
            // 残余ネットワークの更新
            // フローを流せる場合、残余グラフの容量をflowだけ増減させる
            if (flow > 0)
            {
                e.cap -= flow;              // u->vの辺を減少
                G[e.to][e.rev].cap += flow; // v->uの辺を増加
                return flow;
            }
        }
        return 0;
    }

    // 頂点sから頂点tまでの最大フローの総流量を返す
    ll maxFlow(ll s, ll t)
    {
        ll totalFlow = 0;
        while (true)
        {
            // s->tに探索する前に記録した頂点をリセット
            visited.assign(size, false);
            ll F = dfs(s, t, INF); // s->tへの流量を取得
            // フローを流せなくなったら終了
            if (F == 0)
                break;
            totalFlow += F;
        }
        return totalFlow;
    }
};

// グローバル変数で管理
ll n, m;
ll a, b, c;

// FordFulkersonのインスタンスff
FordFulkerson ff;

int main() {
    ll N, M;
    cin >> N >> M;
    ll D;
    cin >> D;
    vector<vector<ll>> li(M, vector<ll>(5));
    set<ll> st;
    rep(i, 0, M) {
        rep(j, 0, 5) {
            cin >> li.at(i).at(j);
        }
        li.at(i).at(0)--;
        li.at(i).at(1)--;
        li.at(i).at(3)+=D;
        st.insert(li.at(i).at(2));
        st.insert(li.at(i).at(3));
    }
    st.insert(0);
    st.insert(2000000000);
    map<ll, ll> G;
    ll now=0;
    while(st.size()) {
        G[*begin(st)]=now;
        st.erase(*begin(st));
        now++;
    }
    n=N*now;
    rep(i, 0, M) {
        rep(j, 2, 4) {
            li.at(i).at(j)=G[li.at(i).at(j)];
        }
    }
    ff.init(n);
    rep(i, 0, N) {
        rep(j, 0, now-1) {
            ff.add_edge(i*now+j, i*now+j+1, linf);
        }
    }
    rep(i, 0, M) {
        ff.add_edge(li.at(i).at(0)*now+li.at(i).at(2), li.at(i).at(1)*now+li.at(i).at(3), li.at(i).at(4));
    }
    cout << ff.maxFlow(0, now*N-1);
}
0