// ---------- begin max flow (Dinic) ---------- mod maxflow { pub trait MaxFlowCapacity: Copy + Ord + std::ops::Add + std::ops::Sub { fn zero() -> Self; fn inf() -> Self; } macro_rules! impl_primitive_integer_capacity { ($x:ty) => { impl MaxFlowCapacity for $x { fn zero() -> Self { 0 } fn inf() -> Self { <$x>::MAX } } }; } impl_primitive_integer_capacity!(u32); impl_primitive_integer_capacity!(u64); impl_primitive_integer_capacity!(i32); impl_primitive_integer_capacity!(i64); #[derive(Clone, Copy)] struct Edge { to_: u32, inv_: u32, cap_: Cap, } impl Edge { fn new(to: usize, inv: usize, cap: Cap) -> Self { Edge { to_: to as u32, inv_: inv as u32, cap_: cap, } } fn to(&self) -> usize { self.to_ as usize } fn inv(&self) -> usize { self.inv_ as usize } } impl Edge { fn add(&mut self, cap: Cap) { self.cap_ = self.cap_ + cap; } fn sub(&mut self, cap: Cap) { self.cap_ = self.cap_ - cap; } fn cap(&self) -> Cap { self.cap_ } } pub struct Graph { graph: Vec>>, } #[allow(dead_code)] pub struct EdgeIndex { src: usize, dst: usize, x: usize, y: usize, } impl Graph { pub fn new(size: usize) -> Self { Self { graph: vec![vec![]; size], } } pub fn add_edge(&mut self, src: usize, dst: usize, cap: Cap) -> EdgeIndex { assert!(src.max(dst) < self.graph.len()); assert!(cap >= Cap::zero()); assert!(src != dst); let x = self.graph[src].len(); let y = self.graph[dst].len(); self.graph[src].push(Edge::new(dst, y, cap)); self.graph[dst].push(Edge::new(src, x, Cap::zero())); EdgeIndex { src, dst, x, y } } // src, dst, used, intial_capacity #[allow(dead_code)] pub fn get_edge(&self, e: &EdgeIndex) -> (usize, usize, Cap, Cap) { let max = self.graph[e.src][e.x].cap() + self.graph[e.dst][e.y].cap(); let used = self.graph[e.dst][e.y].cap(); (e.src, e.dst, used, max) } pub fn flow(&mut self, src: usize, dst: usize) -> Cap { let size = self.graph.len(); assert!(src.max(dst) < size); assert!(src != dst); let mut queue = std::collections::VecDeque::new(); let mut level = vec![0; size]; let mut it = vec![0; size]; let mut ans = Cap::zero(); loop { queue.clear(); level.clear(); level.resize(size, 0); level[src] = 1; queue.push_back(src); while let Some(v) = queue.pop_front() { if v == dst { break; } let d = level[v] + 1; for e in self.graph[v].iter() { let u = e.to(); if e.cap() > Cap::zero() && level[u] == 0 { level[u] = d; queue.push_back(u); } } } if level[dst] == 0 { break; } it.clear(); it.resize(size, 0); loop { let f = self.dfs(dst, src, Cap::inf(), &mut it, &level); if f == Cap::zero() { break; } ans = ans + f; } } ans } fn dfs(&mut self, v: usize, src: usize, cap: Cap, it: &mut [usize], level: &[u32]) -> Cap { if v == src { return cap; } while it[v] < self.graph[v].len() { let (u, inv) = { let po = &self.graph[v][it[v]]; (po.to(), po.inv()) }; if level[u] + 1 == level[v] && self.graph[u][inv].cap() > Cap::zero() { let cap = std::cmp::min(cap, self.graph[u][inv].cap()); let c = self.dfs(u, src, cap, it, level); if c > Cap::zero() { self.graph[v][it[v]].add(c); self.graph[u][inv].sub(c); return c; } } it[v] += 1; } Cap::zero() } } } // ---------- end max flow (Dinic) ---------- // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let s = { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); s }; let mut iter = s.split_whitespace(); input_inner!{iter, $($r)*} }; } macro_rules! input_inner { ($iter:expr) => {}; ($iter:expr, ) => {}; ($iter:expr, $var:ident : $t:tt $($r:tt)*) => { let $var = read_value!($iter, $t); input_inner!{$iter $($r)*} }; } macro_rules! read_value { ($iter:expr, ( $($t:tt),* )) => { ( $(read_value!($iter, $t)),* ) }; ($iter:expr, [ $t:tt ; $len:expr ]) => { (0..$len).map(|_| read_value!($iter, $t)).collect::>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::>() }; ($iter:expr, bytes) => { read_value!($iter, String).bytes().collect::>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } // ---------- end input macro ---------- fn run() { input! { h: usize, w: usize, g: [[i64; w]; h], r: [i64; h], c: [i64; w], } let mut graph = maxflow::Graph::new(h * w + h + w + 2); let src = h * w + h + w; let dst = src + 1; let sum = r.iter().chain(c.iter()).sum::(); for (i, g) in g.iter().enumerate() { for (j, g) in g.iter().enumerate() { graph.add_edge(i * w + j, dst, *g); } } let inf = 10i64.pow(9) * 300 + 1; for (i, r) in r.iter().enumerate() { graph.add_edge(src, h * w + i, *r); for j in 0..w { graph.add_edge(h * w + i, i * w + j, inf); } } for (j, c) in c.iter().enumerate() { graph.add_edge(src, h * w + h + j, *c); for i in 0..h { graph.add_edge(h * w + h + j, i * w + j, inf); } } let ans = sum - graph.flow(src, dst); println!("{}", ans); } fn main() { run(); }