結果

問題 No.1282 Display Elements
ユーザー Strorkis
提出日時 2020-11-06 22:15:05
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 2,253 bytes
コンパイル時間 13,092 ms
コンパイル使用メモリ 377,608 KB
実行使用メモリ 6,608 KB
最終ジャッジ日時 2024-07-22 13:01:01
合計ジャッジ時間 14,902 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

pub mod procon_input {
    use std::{str::FromStr, iter::FromIterator};

    pub fn read_line() -> String {
        let mut input = String::new();
        std::io::stdin().read_line(&mut input).ok();
        input
    }

    pub fn parse<T: FromStr>(s: &str) -> T {
        s.parse().ok().unwrap()
    }
    
    pub fn read<T: FromStr>() -> T {
        parse(read_line().trim_end())
    }
    
    pub fn read_collection<T: FromStr, C: FromIterator<T>>() -> C {
        read_line().split_whitespace().map(parse).collect()
    }

    #[macro_export]
    macro_rules! read_tuple {
        ( $( $t:ty ),* ) => {{
            let input = read_line();
            let mut iter = input.split_whitespace();
            ( $( parse::<$t>(iter.next().unwrap()) ),* )
        }};
    }
}

struct FenwickTree<T> {
    n: usize,
    init: T,
    data: Vec<T>,
}

impl<T: Copy + std::ops::AddAssign> FenwickTree<T> {
    fn new(n: usize, init: T) -> FenwickTree<T> {
        FenwickTree { n, init, data: vec![init; n] }
    }

    fn add(&mut self, mut p: usize, x: T) {
        p += 1;
        while p <= self.n {
            self.data[p - 1] += x;
            p += p & (!p + 1);
        }
    }

    fn sum(&self, mut r: usize) -> T {
        let mut res = self.init;
        while r > 0 {
            res += self.data[r - 1];
            r -= r & (!r + 1);
        }
        res
    }
}

use procon_input::*;

fn solve(writer: &mut std::io::BufWriter<std::io::StdoutLock>) {
    use std::io::Write;

    let n: usize = read();
    let mut a: Vec<i32> = read_collection();
    a.sort();
    let b: Vec<i32> = read_collection();

    let mut v: Vec<i32> = a.iter().chain(b.iter()).cloned().collect();
    v.sort();
    v.dedup();

    let a: Vec<usize> = {
        a.into_iter().map(|x| v.binary_search(&x).unwrap()).collect()
    };
    let b: Vec<usize> = {
        b.into_iter().map(|x| v.binary_search(&x).unwrap()).collect()
    };

    let mut ans: i64 = 0;
    let mut ft = FenwickTree::new(2 * n, 0);
    for i in 0..n {
        ft.add(b[i], 1);
        ans += ft.sum(a[i]);
    }
    writeln!(writer, "{}", ans).ok();
}

fn main() {
    let stdout = std::io::stdout();
    let mut writer = std::io::BufWriter::new(stdout.lock());
    solve(&mut writer);
}
0