#662807 (Rust) No.957 植林

• 問題: No.957 植林
• ユーザー: SSlime
• 提出日時: 2021-06-03 20:23:56
• 言語: Rust (1.77.0)
• 結果: AC
• 実行時間: 666 ms / 2,000 ms
• コード長: 4,647 bytes
• コンパイル時間: 11,849 ms
• コンパイル使用メモリ: 378,176 KB
• 実行使用メモリ: 10,752 KB
• 最終ジャッジ日時: 2024-04-28 14:49:40
• 合計ジャッジ時間: 27,633 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	AC	16 ms; 9,088 KB
testcase_04	AC	16 ms; 8,576 KB
testcase_05	AC	15 ms; 9,216 KB
testcase_06	AC	19 ms; 9,600 KB
testcase_07	AC	16 ms; 8,704 KB
testcase_08	AC	10 ms; 9,088 KB
testcase_09	AC	10 ms; 9,088 KB
testcase_10	AC	11 ms; 9,472 KB
testcase_11	AC	11 ms; 9,216 KB
testcase_12	AC	9 ms; 8,960 KB
testcase_13	AC	7 ms; 7,296 KB
testcase_14	AC	9 ms; 9,600 KB
testcase_15	AC	9 ms; 8,960 KB
testcase_16	AC	8 ms; 7,040 KB
testcase_17	AC	9 ms; 8,448 KB
testcase_18	AC	461 ms; 8,832 KB
testcase_19	AC	479 ms; 9,088 KB
testcase_20	AC	491 ms; 9,472 KB
testcase_21	AC	535 ms; 9,216 KB
testcase_22	AC	555 ms; 9,472 KB
testcase_23	AC	568 ms; 9,600 KB
testcase_24	AC	591 ms; 9,856 KB
testcase_25	AC	666 ms; 10,240 KB
testcase_26	AC	625 ms; 10,240 KB
testcase_27	AC	639 ms; 10,240 KB
testcase_28	AC	631 ms; 10,240 KB
testcase_29	AC	640 ms; 10,240 KB
testcase_30	AC	660 ms; 10,240 KB
testcase_31	AC	489 ms; 8,960 KB
testcase_32	AC	487 ms; 9,216 KB
testcase_33	AC	516 ms; 9,344 KB
testcase_34	AC	529 ms; 9,216 KB
testcase_35	AC	545 ms; 9,472 KB
testcase_36	AC	571 ms; 9,600 KB
testcase_37	AC	578 ms; 9,856 KB
testcase_38	AC	643 ms; 10,368 KB
testcase_39	AC	632 ms; 10,240 KB
testcase_40	AC	623 ms; 10,240 KB
testcase_41	AC	8 ms; 9,984 KB
testcase_42	AC	8 ms; 9,984 KB
testcase_43	AC	12 ms; 10,752 KB
testcase_44	AC	13 ms; 10,752 KB
testcase_45	AC	0 ms; 5,376 KB
testcase_46	AC	1 ms; 5,376 KB
testcase_47	AC	1 ms; 5,376 KB

ソースコード

// -*- coding:utf-8-unix -*-

use proconio::input;

use max_flow::MfGraph;

#[allow(dead_code)]
mod max_flow {
    struct _Edge {
        to: usize,
        rev: usize,
        cap: u64,
    }
    pub struct MfGraph {
        n: usize,
        pos: Vec<(usize, usize)>,
        g: Vec<Vec<_Edge>>
    }
    impl MfGraph {
        pub fn new(n: usize) -> MfGraph {
            MfGraph {
                n,
                pos: Vec::new(),
                g: (0..n).map(|_| Vec::new()).collect(),
            }
        }
        pub fn add_edge(&mut self, from: usize, to: usize, cap: u64) {
            let rev = self.g[to].len();
            self.g[from].push(_Edge {to, cap, rev});
            let rev = self.g[from].len() - 1;
            self.g[to].push(_Edge {to: from, cap: 0, rev});
        }
        pub fn flow(&mut self, s: usize, t: usize) -> u64 {
            self.flow_with_cap(s, t, u64::max_value())
        }
        pub fn flow_with_cap(&mut self, s: usize, t: usize, limit: u64) -> u64 {
            assert!(s < self.n && t < self.n);
            let n = self.n;
            let mut solver = FlowSolver {
                graph: self,
                s,
                t,
                limit,
                level: vec![0; n],
                iter: vec![0; n],
            };
            let mut flow = 0;
            while flow < limit {
                solver.bfs();
                if solver.level[t] == -1 {
                    break;
                }
                solver.iter.iter_mut().for_each(|e| *e = 0);
                while flow < limit {
                    let f = solver.dfs(s, limit - flow);
                    if f == 0 {
                        break;
                    }
                    flow += f;
                }
            }
            flow
        }
        pub fn min_cut(&mut self, s: usize) -> Vec<bool> {
            let mut visited = vec![false; self.n];
            let mut que = std::collections::VecDeque::new();
            que.push_back(s);
            while let Some(p) = que.pop_front() {
                visited[p] = true;
                for e in &self.g[p] {
                    if e.cap != 0 && !visited[e.to] {
                        visited[e.to] = true;
                        que.push_back(e.to);
                    }
                }
            }
            visited
        }
    }
    struct FlowSolver<'a> {
        graph: &'a mut MfGraph,
        s: usize,
        t: usize,
        limit: u64,
        level: Vec<i32>,
        iter: Vec<usize>,
    }
    impl FlowSolver<'_> {
        fn bfs(&mut self) {
            self.level.iter_mut().for_each(|e| *e = -1);
            self.level[self.s] = 0;
            let mut que = std::collections::VecDeque::<usize>::new();
            que.push_back(self.s);
            while let Some(now) = que.pop_front() {
                for &_Edge {to, cap, ..} in &self.graph.g[now] {
                    if cap <= 0 || self.level[to] >= 0 {continue;}
                    self.level[to] = self.level[now] + 1;
                    if self.t == to {
                        return;
                    }
                    que.push_back(to);
                }
            }
        }
        fn dfs(&mut self, v: usize, up: u64) -> u64 {
            if v == self.t {
                return up;
            }
            for i in self.iter[v]..self.graph.g[v].len() {
                self.iter[v] = i;
                let &_Edge {
                    to,
                    rev,
                    cap
                } = &self.graph.g[v][i];
                if cap <= 0 || self.level[v] >= self.level[to] {
                    continue;
                }
                let d = self.dfs(to, up.min(cap));
                if d <= 0 {
                    continue;
                }
                self.graph.g[v][i].cap -= d;
                self.graph.g[to][rev].cap += d;
                return d;
            }
            self.iter[v] = self.graph.g[v].len();
            0
        }
    }
}

fn main() {
    input! {
        h: usize,
        w: usize,
        g: [[i64; w]; h],
        r: [i64; h],
        c: [i64; w],
    }
    let mut graph = MfGraph::new(h+w+2);
    let mut s = 0;
    for i in 0..h {
        graph.add_edge(0, i+1, (&g[i]).into_iter().sum::<i64>() as u64);
        graph.add_edge(i+1, h+w+1, r[i] as u64);
        s += r[i] as u64;
    }
    for j in 0..w {
        graph.add_edge(h+j+1, h+w+1, c[j] as u64);
        s += c[j] as u64;
        for i in 0..h {
            graph.add_edge(i+1, h+j+1, g[i][j] as u64);
        }
    }
    let f = graph.flow(0, h+w+1);
    println!("{}", s - f);
}