Optimize min_plus_conv

koba-e964 · Apr 18, 2024 · 263ce45 · 263ce45
1 parent c2acca7
commit 263ce45
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/atcoder/abc348/g.rs b/atcoder/abc348/g.rs
@@ -68,13 +68,13 @@ fn min_plus_conv(a: &[i64], b: &[i64]) -> Vec<i64> {
     }
     let mut dp = vec![INF; n + m - 1];
     let cost_fun = |j: usize, i: usize| {
-        if i >= j && j < n && i < j + m {
-            a[j] + b[i - j]
+        if i >= j && i < j + m {
+            b[i - j]
         } else {
             INF / 2
         }
     };
-    monotone_minima(0, n, 0, n + m - 1, &vec![0; n], &mut dp, &cost_fun);
+    monotone_minima(0, n, 0, n + m - 1, &a, &mut dp, &cost_fun);
     dp
 }
 

diff --git a/judge.yosupo.jp/min_plus_convolution_convex_arbitrary.rs b/judge.yosupo.jp/min_plus_convolution_convex_arbitrary.rs
@@ -68,13 +68,13 @@ fn min_plus_conv(a: &[i64], b: &[i64]) -> Vec<i64> {
     }
     let mut dp = vec![INF; n + m - 1];
     let cost_fun = |j: usize, i: usize| {
-        if i >= j && j < n && i < j + m {
-            a[j] + b[i - j]
+        if i >= j && i < j + m {
+            b[i - j]
         } else {
             INF / 2
         }
     };
-    monotone_minima(0, n, 0, n + m - 1, &vec![0; n], &mut dp, &cost_fun);
+    monotone_minima(0, n, 0, n + m - 1, &a, &mut dp, &cost_fun);
     dp
 }