結果

問題 No.1606 Stuffed Animals Keeper
コンテスト
ユーザー phspls
提出日時 2022-10-26 01:13:48
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 65 ms / 3,000 ms
コード長 1,235 bytes
コンパイル時間 13,425 ms
コンパイル使用メモリ 382,388 KB
実行使用メモリ 67,200 KB
最終ジャッジ日時 2024-07-03 20:30:45
合計ジャッジ時間 14,740 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

const INF: usize = 1usize << 60;

fn main() {
    let mut n = String::new();
    std::io::stdin().read_line(&mut n).ok();
    let n: usize = n.trim().parse().unwrap();
    let mut a = String::new();
    std::io::stdin().read_line(&mut a).ok();
    let a: Vec<_> = a.trim().split_whitespace().collect::<Vec<_>>().join("").split('2').map(|s| s.chars().collect::<Vec<_>>()).collect();

    let zeros = a.iter().filter(|&v| v.len() > 0).map(|v| v.iter().filter(|&&c| c == '1').count()).sum::<usize>();
    let one_cnt = a.iter().filter(|&v| v.len() > 0).map(|v| v.iter().filter(|&&c| c == '1').count()).collect::<Vec<_>>();
    let size_cnt = a.iter().filter(|&v| v.len() > 0).map(|v| v.len()).collect::<Vec<_>>();
    let mut dp = vec![vec![INF; n+1]; size_cnt.len()+1];
    dp[0][0] = 0;
    for i in 0..size_cnt.len() {
        let to_zero_cnt = size_cnt[i] - one_cnt[i];
        for j in 0..=n {
            if dp[i][j] == INF { continue; }
            dp[i+1][j+size_cnt[i]] = dp[i+1][j+size_cnt[i]].min(dp[i][j] + to_zero_cnt);
            dp[i+1][j] = dp[i+1][j].min(dp[i][j]);
        }
    }
    if dp[size_cnt.len()][zeros] == INF {
        println!("-1");
    } else {
        println!("{}", dp[size_cnt.len()][zeros]);
    }
}
0