結果

問題 No.275 中央値を求めよ
ユーザー CosnomiCosnomi
提出日時 2019-07-15 03:17:17
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 2,074 bytes
コンパイル時間 16,792 ms
コンパイル使用メモリ 395,836 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-25 04:14:00
合計ジャッジ時間 15,293 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 4 RE * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

macro_rules! get {
    ($t:ty) => {
        {
            let mut line: String = String::new();
            std::io::stdin().read_line(&mut line).unwrap();
            line.trim().parse::<$t>().unwrap()
        }
    };
    ($($t:ty),*) => {
        {
            let mut line: String = String::new();
            std::io::stdin().read_line(&mut line).unwrap();
            let mut iter = line.split_whitespace();
            (
                $(iter.next().unwrap().parse::<$t>().unwrap(),)*
            )
        }
    };
    ($t:ty; $n:expr) => {
        (0..$n).map(|_|
            get!($t)
        ).collect::<Vec<_>>()
    };
    ($($t:ty),*; $n:expr) => {
        (0..$n).map(|_|
            get!($($t),*)
        ).collect::<Vec<_>>()
    };
    ($t:ty ;;) => {
        {
            let mut line: String = String::new();
            std::io::stdin().read_line(&mut line).unwrap();
            line.split_whitespace()
                .map(|t| t.parse::<$t>().unwrap())
                .collect::<Vec<_>>()
        }
    };
    ($t:ty ;; $n:expr) => {
        (0..$n).map(|_| get!($t ;;)).collect::<Vec<_>>()
    };
}

fn q_sort(nums: Vec<u32>) -> Vec<u32> {
    if nums.len() <= 1 {
        nums
    } else {
        let pivot = nums[0];
        let mut left: Vec<u32> = Vec::new();
        let mut right: Vec<u32> = Vec::new();
        for num in (nums[1..]).into_iter() {
            if *num <= pivot {
                left.push(*num)
            } else {
                right.push(*num)
            }
        }
        let mut sorted_l = q_sort(left);
        let mut sorted_r = q_sort(right);
        let mut res = Vec::new();
        res.append(&mut sorted_l);
        res.push(pivot);
        res.append(&mut sorted_r);
        res
    }
}

fn get_median(nums: Vec<u32>) -> f32 {
    let len = nums.len();
    if len % 2 == 0 {
        (nums[len / 2 - 1] + nums[len / 2]) as f32 / 2.0
    } else {
        nums[len / 2] as f32
    }
}

fn main() {
    let _len = get!(u32);
    let data: Vec<u32> = get!(u32;;);
    println!("{}", get_median(q_sort(data)))
}
0