結果

問題 No.2028 Even Choice
ユーザー phsplsphspls
提出日時 2022-09-01 00:13:42
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 970 bytes
コンパイル時間 14,321 ms
コンパイル使用メモリ 377,548 KB
実行使用メモリ 8,832 KB
最終ジャッジ日時 2024-11-08 18:44:34
合計ジャッジ時間 16,620 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused imports: `BTreeMap`, `BinaryHeap`
 --> src/main.rs:1:25
  |
1 | use std::{collections::{BTreeMap, BinaryHeap}, cmp::Reverse};
  |                         ^^^^^^^^  ^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` on by default

ソースコード

diff # 

use std::{collections::{BTreeMap, BinaryHeap}, cmp::Reverse};


fn main() {
    let mut nk = String::new();
    std::io::stdin().read_line(&mut nk).ok();
    let nk: Vec<usize> = nk.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let n = nk[0];
    let k = nk[1];
    let mut a = String::new();
    std::io::stdin().read_line(&mut a).ok();
    let a: Vec<usize> = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();

    let mut result = 0usize;
    let mut minidx = n;
    let mut a = a.into_iter().skip(1).enumerate().collect::<Vec<(usize, usize)>>();
    a.sort_by_key(|&(i, v)| (Reverse(v), i%2, i));
    for i in 0..k-1 {
        let (idx, val) = a[i];
        minidx = idx.min(minidx);
        result += val;
    }
    for i in k-1..n-1 {
        let (idx, val) = a[i];
        let minidx = idx.min(minidx);
        if minidx % 2 == 0 {
            result += val;
            break;
        }
    }
    println!("{}", result);
}
0