結果

問題 No.2740 Old Maid
ユーザー Yukino DX.
提出日時 2024-04-27 15:49:02
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 71 ms / 2,000 ms
コード長 1,949 bytes
コンパイル時間 25,575 ms
コンパイル使用メモリ 380,372 KB
実行使用メモリ 15,812 KB
最終ジャッジ日時 2024-11-15 19:17:33
合計ジャッジ時間 33,951 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

#![allow(unused_imports)]
//use itertools::{iproduct, Itertools};
use proconio::input;
use proconio::marker::*;
use std::collections::*;

fn main() {
    input! {
        n:usize,
        p:[Usize1;n],
    }

    let mut nexts = vec![None; n];
    let mut prevs = vec![None; n];
    for i in 0..n {
        if i != n - 1 {
            nexts[p[i]] = Some(p[i + 1]);
        }
        if i != 0 {
            prevs[p[i]] = Some(p[i - 1]);
        }
    }

    let mut rem = (0..n).collect::<BTreeSet<_>>();
    let mut ans = vec![];
    while !rem.is_empty() {
        let fst = *rem.first().unwrap();
        if let Some(nxt) = nexts[fst] {
            ans.push(fst);
            ans.push(nxt);
            match (prevs[fst], nexts[nxt]) {
                (Some(pre), Some(nnxt)) => {
                    nexts[pre] = Some(nnxt);
                    prevs[nnxt] = Some(pre);
                }
                (Some(pre), None) => {
                    nexts[pre] = None;
                }
                (None, Some(nnxt)) => {
                    prevs[nnxt] = None;
                }
                _ => (),
            }

            rem.remove(&fst);
            rem.remove(&nxt);
            continue;
        }

        let scn = *rem.iter().nth(1).unwrap();
        if let Some(nxt) = nexts[scn] {
            ans.push(scn);
            ans.push(nxt);

            match (prevs[scn], nexts[nxt]) {
                (Some(pre), Some(nnxt)) => {
                    nexts[pre] = Some(nnxt);
                    prevs[nnxt] = Some(pre);
                }
                (Some(pre), None) => {
                    nexts[pre] = None;
                }
                (None, Some(nnxt)) => {
                    prevs[nnxt] = None;
                }
                _ => (),
            }

            rem.remove(&scn);
            rem.remove(&nxt);
        }
    }

    for i in 0..n {
        print!("{} ", ans[i] + 1);
    }
    println!();
}
0