結果

問題 No.1823 Tricolor Dango
ユーザー zeronosu77108
提出日時 2022-01-28 22:15:15
言語 Rust
(1.83.0 + proconio)
結果
TLE  
実行時間 -
コード長 2,639 bytes
コンパイル時間 13,953 ms
コンパイル使用メモリ 383,412 KB
実行使用メモリ 15,688 KB
最終ジャッジ日時 2024-12-30 07:36:20
合計ジャッジ時間 84,383 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample TLE * 1
other AC * 4 TLE * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::BinaryHeap;

fn main() {
    let t : usize = input();
    for _ in 0..t { solve() }
}

fn solve() {
    let _ : usize = input();
    let a : Vec<usize> = input_vec();

    let mut pq : BinaryHeap<_> = a.into_iter().collect();


    while pq.len() >= 3 {
        let mut a = pq.pop().unwrap();
        let mut b = pq.pop().unwrap();
        let mut c = pq.pop().unwrap();

        let mut d = a.min(b).min(c);
        let cnt = vec![a,b,c].iter().filter(|&&c| c==d).count();
        if cnt == 2 { d -= 1; }
        a -= d; b -= d; c -= d;
        if a > 0 { pq.push(a); }
        if b > 0 { pq.push(b); }
        if c > 0 { pq.push(c); }
    }

    println!("{}", if pq.is_empty() {"Yes"} else {"No"});
}

#[allow(dead_code)]
fn input<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

#[allow(dead_code)]
fn input_t<T: std::str::FromStr, U: std::str::FromStr>() -> (T, U) {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    let s = s.trim().split_whitespace().collect::<Vec<&str>>();
    (s[0].parse().ok().unwrap(), s[1].parse().ok().unwrap())
}

#[allow(dead_code)]
fn input_t3<T1: std::str::FromStr, T2: std::str::FromStr, T3: std::str::FromStr>() -> (T1, T2, T3) {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    let s = s.trim().split_whitespace().collect::<Vec<&str>>();
    (s[0].parse().ok().unwrap(), s[1].parse().ok().unwrap(), s[2].parse().ok().unwrap())
}

#[allow(dead_code)]
fn input_t4<T1: std::str::FromStr, T2: std::str::FromStr, T3: std::str::FromStr, T4: std::str::FromStr>() -> (T1, T2, T3, T4) {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    let s = s.trim().split_whitespace().collect::<Vec<&str>>();
    (s[0].parse().ok().unwrap(), s[1].parse().ok().unwrap(), s[2].parse().ok().unwrap(), s[3].parse().ok().unwrap())
}

#[allow(dead_code)]
fn input_t5<T1: std::str::FromStr, T2: std::str::FromStr, T3: std::str::FromStr, T4: std::str::FromStr, T5: std::str::FromStr>() -> (T1, T2, T3, T4, T5) {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    let s = s.trim().split_whitespace().collect::<Vec<&str>>();
    (s[0].parse().ok().unwrap(), s[1].parse().ok().unwrap(), s[2].parse().ok().unwrap(), s[3].parse().ok().unwrap(), s[4].parse().ok().unwrap())
}

#[allow(dead_code)]
fn input_vec<T: std::str::FromStr>() -> Vec<T> {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().split_whitespace().map(|s| s.parse().ok().unwrap()).collect()
}
0