結果

問題 No.957 植林
ユーザー shin3110
提出日時 2023-11-12 19:18:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 346 ms / 2,000 ms
コード長 2,595 bytes
コンパイル時間 3,981 ms
コンパイル使用メモリ 240,184 KB
実行使用メモリ 9,020 KB
最終ジャッジ日時 2024-09-26 03:03:50
合計ジャッジ時間 13,227 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;

#define ll long long

struct edge {
    int to, seq;
    ll cup;
};

class max_flow {
    public:
    int siz;
    vector<vector<edge>> side;
    vector<int> dist;

    max_flow (int n) {
        siz = n;
        side.assign(siz, {});
        return ;
    }

    void set(int pos, int to, ll x) {
        int a = side[to].size();
        int b = side[pos].size();
        side[pos].push_back(edge{to, a, x});
        side[to].push_back(edge{pos, b, 0});
        return ;
    }

    void bfs(int s) {
        dist.assign(siz, -1);
        dist[s] = 0;
        queue<int> q;
        q.push(s);
        while(!q.empty()) {
            int pos = q.front(); q.pop();
            for(int i=0; i<side[pos].size(); i++) {
                edge e = side[pos][i];
                if(e.cup <= 0) continue;
                if(dist[e.seq] >= 0) continue;
                dist[e.seq] = dist[pos]+1;
                q.push(e.seq);
            }
        }
    }

    ll dfs(int pos, int goal, ll F) {
        if(pos == goal) return F;
        return 0;
    }

    ll len(int s, int t) {
        ll res = 0;
        while(1) {
            bfs(s);
            if(dist[t] < 0) return res;
        }
        return res;
    }
};

/*
signed main() {
    int h, w; cin >> h >> w;
    int ax = h+w+1;
    max_flow F(ax+1);
    vector<ll> c(h+1, 0);
    for(int i=0; i<h*w; i++) {
        ll g; cin >> g;
        int x = i/w;
        c[x+1] += g;
        int y = i%w;
        F.set(x+1, h+y+1, g);
    }
    ll ans = 0;
    for(int j=1; j<=h; j++) {
        ll r; cin >> r;
        r -= c[j];
        if(r < 0) F.set(0, j, -r);
        else {
            ans += r;
            F.set(j, ax, r);
        }
    }
    for(int i=1; i<=w; i++) {
        ll r; cin >> r;
        F.set(h+i, ax, r);
        ans += r;
    }
    ans -= F.len(0, ax);
    cout << ans << endl;
} */

signed main() {
    int h, w; cin >> h >> w;
    int ax = h+w+1;
    mf_graph<ll> F(ax+1);
    vector<ll> c(h+1, 0);
    for(int i=0; i<h*w; i++) {
        ll g; cin >> g;
        int x = i/w;
        c[x+1] += g;
        int y = i%w;
        F.add_edge(x+1, h+y+1, g);
    }
    ll ans = 0;
    for(int j=1; j<=h; j++) {
        ll r; cin >> r;
        r -= c[j];
        if(r < 0) F.add_edge(0, j, -r);
        else {
            ans += r;
            F.add_edge(j, ax, r);
        }
    }
    for(int i=1; i<=w; i++) {
        ll r; cin >> r;
        F.add_edge(h+i, ax, r);
        ans += r;
    }
    ans -= F.flow(0, ax);
    cout << ans << endl;
}
0