結果

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

ソースコード

diff # 

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

#define ll long long

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

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

    max_flow (int n) {
        siz = n;
        side.assign(siz, {});
        used.assign(siz, false);
        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 ;
    }

    ll dfs(int pos, int goal, ll F) {
        if(pos == goal) return F;
        used[pos] = true;
        for(int i=0; i<side[pos].size(); i++) {
            if(side[pos][i].cup == 0) continue;
            if(used[side[pos][i].to]) continue;
            ll G = dfs(side[pos][i].to, goal, min(F, side[pos][i].cup));
            if(G > 0) {
                side[pos][i].cup -= G;
                side[side[pos][i].to][side[pos][i].seq].cup += G;
                return G;
            }
        }
        return 0;
    }

    ll len(int s, int t) {
        ll res = 0;
        while(1) {
            for(int i=0; i<siz; i++) used[i] = false;
            ll F = dfs(s, t, 1e18);
            if(F == 0) break;
            res += F;
        }
        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;
}
0