結果
問題 | No.2423 Merge Stones |
ユーザー |
![]() |
提出日時 | 2023-08-13 10:23:25 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 443 ms / 4,000 ms |
コード長 | 2,615 bytes |
コンパイル時間 | 12,262 ms |
コンパイル使用メモリ | 403,556 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-21 04:18:10 |
合計ジャッジ時間 | 34,813 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 72 |
ソースコード
fn main() {input! {n: usize,k: usize,a: [i64; n],c: [usize1; n],}let mut sum = a.iter().cloned().cycle().take(2 * n).collect::<Vec<_>>();sum.push(0);for i in (0..(2 * n)).rev() {sum[i] += sum[i + 1];}let mut dp = vec![vec![0usize; 2 * n + 1]; 2 * n + 1];for (i, c) in c.iter().enumerate() {dp[i][i + 1] = 1 << *c;dp[n + i][n + i + 1] = 1 << *c;}let mut ans = *a.iter().max().unwrap();for len in 2..=n {for l in 0..n {let r = l + len;let mut v = 0;for m in (l + 1)..r {for j in 0..=k {v |= dp[l][m] & (dp[m][r] << j);v |= dp[l][m] & (dp[m][r] >> j);v |= (dp[l][m] >> j) & dp[m][r];v |= (dp[l][m] << j) & dp[m][r];}}dp[l][r] = v;if r + n <= 2 * n {dp[l + n][r + n] = v;}if v != 0 {ans = ans.max(sum[l] - sum[r]);}}}println!("{}", ans);}// ---------- begin input macro ----------// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8#[macro_export]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_export]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_export]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::<Vec<_>>()};($iter:expr, chars) => {read_value!($iter, String).chars().collect::<Vec<char>>()};($iter:expr, bytes) => {read_value!($iter, String).bytes().collect::<Vec<u8>>()};($iter:expr, usize1) => {read_value!($iter, usize) - 1};($iter:expr, $t:ty) => {$iter.next().unwrap().parse::<$t>().expect("Parse error")};}// ---------- end input macro ----------