結果

問題 No.1301 Strange Graph Shortest Path
ユーザー firiexpfiriexp
提出日時 2020-11-27 23:46:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 166 ms / 3,000 ms
コード長 3,135 bytes
コンパイル時間 1,311 ms
コンパイル使用メモリ 116,948 KB
実行使用メモリ 34,988 KB
最終ジャッジ日時 2024-07-26 19:54:03
合計ジャッジ時間 7,333 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 131 ms
33,244 KB
testcase_03 AC 112 ms
29,532 KB
testcase_04 AC 163 ms
31,616 KB
testcase_05 AC 114 ms
32,596 KB
testcase_06 AC 140 ms
29,584 KB
testcase_07 AC 137 ms
31,540 KB
testcase_08 AC 110 ms
29,804 KB
testcase_09 AC 130 ms
28,032 KB
testcase_10 AC 109 ms
29,324 KB
testcase_11 AC 143 ms
30,456 KB
testcase_12 AC 142 ms
30,588 KB
testcase_13 AC 136 ms
33,272 KB
testcase_14 AC 135 ms
28,228 KB
testcase_15 AC 127 ms
29,216 KB
testcase_16 AC 166 ms
31,616 KB
testcase_17 AC 138 ms
33,508 KB
testcase_18 AC 126 ms
30,432 KB
testcase_19 AC 145 ms
29,568 KB
testcase_20 AC 145 ms
28,672 KB
testcase_21 AC 140 ms
31,832 KB
testcase_22 AC 148 ms
29,440 KB
testcase_23 AC 132 ms
32,896 KB
testcase_24 AC 142 ms
29,312 KB
testcase_25 AC 151 ms
31,488 KB
testcase_26 AC 134 ms
30,444 KB
testcase_27 AC 140 ms
30,612 KB
testcase_28 AC 111 ms
32,300 KB
testcase_29 AC 153 ms
30,976 KB
testcase_30 AC 148 ms
31,556 KB
testcase_31 AC 154 ms
31,104 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 79 ms
26,240 KB
testcase_34 AC 153 ms
34,988 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <numeric>
#include <bitset>
#include <cmath>

static const int MOD = 1000000007;
using ll = long long;
using u32 = unsigned;
using u64 = unsigned long long;
using namespace std;

template<class T> constexpr T INF = ::numeric_limits<T>::max() / 32 * 15 + 208;

template<typename F, typename C>
struct PrimalDual {
    struct edge {
        int to; F cap; C cost; int rev;
        edge() = default;
        edge(int to, F cap, C cost, int rev):to(to), cap(cap), cost(cost), rev(rev) {};
    };
    vector<vector<edge>> G;
    vector<C> potential, min_cost;
    vector<int> prevv, preve;

    explicit PrimalDual(int n) : G(n), potential(n), min_cost(n), prevv(n), preve(n) {}

    void add_edge(int u, int v, F cap, C cost){
        G[u].emplace_back(v, cap, cost, G[v].size());
        G[v].emplace_back(u, 0, -cost, G[u].size()-1);
    }

    struct P{
        C first; int second;
        P(C first,int second):first(first),second(second){}
        bool operator<(const P&a) const{return a.first<first;}
    };
    void dijkstra(int s){

        priority_queue<P> Q;
        fill(min_cost.begin(),min_cost.end(), INF<C>);
        min_cost[s] = 0;
        Q.emplace(0, s);
        while(!Q.empty()){
            P p = Q.top(); Q.pop();
            int v = p.second;
            if(min_cost[v] < p.first) continue;
            for(int i = 0; i < G[v].size(); ++i){
                edge &e=G[v][i];
                if(e.cap==0) continue;
                if(min_cost[v]+e.cost+potential[v]-potential[e.to] < min_cost[e.to]){
                    min_cost[e.to] = min_cost[v]+e.cost+potential[v]-potential[e.to];
                    prevv[e.to] = v;
                    preve[e.to] = i;
                    Q.emplace(min_cost[e.to], e.to);
                }
            }
        }
    }

    C flow(int s, int t, F fl, int &ok){
        C res = 0;
        fill(potential.begin(),potential.end(), 0);
        while(fl > 0){
            dijkstra(s);
            if(min_cost[t] == INF<C>){
                ok = 0;
                return res;
            }
            for (int i = 0; i < potential.size(); ++i) {
                if(min_cost[i] < INF<C>) potential[i] += min_cost[i];
            }

            F d = fl;
            for(int v = t; v != s; v = prevv[v]){
                d = min(d, G[prevv[v]][preve[v]].cap);
            }
            fl -= d;
            res += potential[t]*d;
            for(int v = t; v != s; v = prevv[v]){
                G[prevv[v]][preve[v]].cap -= d;
                G[v][G[prevv[v]][preve[v]].rev].cap += d;
            }
        }
        ok = 1;
        return res;
    }
};
int main() {
    int n, m;
    cin >> n >> m;
    PrimalDual<int, ll> G(n);
    for (int i = 0; i < m; ++i) {
        int u, v, c, d;
        scanf("%d %d %d %d", &u, &v, &c, &d);
        u--; v--;
        G.add_edge(u, v, 1, c);
        G.add_edge(v, u, 1, c);
        G.add_edge(u, v, 1, d);
        G.add_edge(v, u, 1, d);
    }
    int ret = 0;
    cout << G.flow(0, n-1, 2, ret) << "\n";
    return 0;
}
0