結果

問題 No.1604 Swap Sort:ONE
ユーザー ixTL255
提出日時 2023-03-02 22:38:58
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,161 bytes
コンパイル時間 13,923 ms
コンパイル使用メモリ 382,920 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 15:31:20
合計ジャッジ時間 15,066 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused variable: `n`
  --> src/main.rs:46:9
   |
46 |     let n: usize = n.trim().parse().unwrap();
   |         ^ help: if this is intentional, prefix it with an underscore: `_n`
   |
   = note: `#[warn(unused_variables)]` on by default

ソースコード

diff #

struct BIT {
    bit: Vec<i64>,
    n: usize,
}

impl BIT {
    fn new(n: usize) -> BIT {
        BIT {
            bit:vec![0; n + 1],
            n: n,
        }
    }

    fn sum(&self, i: usize) -> i64 {
        let mut i = i;
        let mut s: i64 = 0;
        while i > 0 {
            s += self.bit[i];
            i -= (i as i64 & -(i as i64)) as usize;
        }
        s
    }

    fn add(&mut self, i: usize, x: i64) {
        let mut i = i;
        while i <= self.n {
            self.bit[i as usize] += x;
            i += (i as i64 & -(i as i64)) as usize;
        }
    }
 }

fn count_inversion(xs: &Vec<usize>) -> i64 {
    let mut cnt = 0;
    let mut bt = BIT::new(xs.len());
    for i in 0..xs.len() {
        cnt += i as i64 - bt.sum(xs[i]);
        bt.add(xs[i], 1);
    }
    cnt
}

fn main() {
    let mut n = String::new();
    std::io::stdin().read_line(&mut n).ok();
    let n: usize = n.trim().parse().unwrap();
    let mut p = String::new();
    std::io::stdin().read_line(&mut p).ok();
    let p: Vec<usize> = p.trim().split_whitespace()
        .map(|x| x.parse().unwrap()).collect();

    println!("{}", count_inversion(&p));
}
0