結果

問題 No.783 門松計画
コンテスト
ユーザー ngtkana
提出日時 2023-04-30 18:45:40
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1,063 ms / 2,000 ms
コード長 2,895 bytes
コンパイル時間 18,232 ms
コンパイル使用メモリ 389,332 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-14 17:02:06
合計ジャッジ時間 17,527 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

use cmpmore::CmpMore;
use std::{convert::TryFrom, io::stdin, isize::MIN};

fn main() {
    let mut stdin = stdin().lines().map(Result::unwrap);
    let [_n, wmax] = <[_; 2]>::try_from(
        stdin
            .next()
            .unwrap()
            .split_whitespace()
            .map(|x| x.parse::<usize>().unwrap())
            .collect::<Vec<_>>()
            .as_slice(),
    )
    .unwrap();
    let ls = stdin
        .next()
        .unwrap()
        .split_whitespace()
        .map(|x| x.parse::<usize>().unwrap())
        .collect::<Vec<_>>();
    let lmax = ls.iter().max().unwrap() + 1;
    let ws = stdin
        .next()
        .unwrap()
        .split_whitespace()
        .map(|x| x.parse::<usize>().unwrap())
        .collect::<Vec<_>>();
    let mut dp = vec![vec![vec![MIN; lmax + 1]; lmax + 1]; wmax + 1];
    for (&l1, &dw1) in ls.iter().zip(&ws) {
        for (&l2, &dw2) in ls.iter().zip(&ws) {
            for (&l3, &dw3) in ls.iter().zip(&ws) {
                for w in dw1 + dw2 + dw3..=wmax {
                    if (l1 > l2 && l2 < l3 && l1 != l3) || (l1 < l2 && l2 > l3 && l1 != l3) {
                        dp[w - dw1 - dw2 - dw3][l1][l2].change_max((l1 + l2 + l3) as isize);
                    }
                }
            }
        }
    }
    for _ in 0..=wmax {
        let mut swp = dp.clone();
        for (&l, &dw) in ls.iter().zip(&ws) {
            for w in dw..=wmax {
                for l1 in 0..=lmax {
                    for l2 in 0..=lmax {
                        if (l2 > l1 && l1 < l && l2 != l) || (l2 < l1 && l1 > l && l2 != l) {
                            swp[w - dw][l][l1].change_max(dp[w][l1][l2] + l as isize);
                        }
                    }
                }
            }
        }
        dp = swp;
    }
    let ans = (*dp.iter().flatten().flatten().max().unwrap()).max(0);
    println!("{}", ans);
}

// cmpmore {{{
#[allow(dead_code)]
mod cmpmore {
    pub fn change_min<T: PartialOrd>(lhs: &mut T, rhs: T) {
        if *lhs > rhs {
            *lhs = rhs;
        }
    }
    pub fn change_max<T: PartialOrd>(lhs: &mut T, rhs: T) {
        if *lhs < rhs {
            *lhs = rhs;
        }
    }
    #[macro_export]
    macro_rules! change_min {
        ($lhs:expr, $rhs:expr) => {
            let rhs = $rhs;
            let lhs = $lhs;
            $crate::cmpmore::change_min(lhs, rhs);
        };
    }
    #[macro_export]
    macro_rules! change_max {
        ($lhs:expr, $rhs:expr) => {
            let rhs = $rhs;
            let lhs = $lhs;
            $crate::cmpmore::change_max(lhs, rhs);
        };
    }
    pub trait CmpMore: PartialOrd + Sized {
        fn change_min(&mut self, rhs: Self) {
            change_min(self, rhs)
        }
        fn change_max(&mut self, rhs: Self) {
            change_max(self, rhs)
        }
    }
    impl<T: PartialOrd + Sized> CmpMore for T {}
}
// }}}
0