結果

問題 No.59 鉄道の旅
ユーザー mio_hmio_h
提出日時 2017-02-20 22:30:27
言語 Rust
(1.72.1)
結果
AC  
実行時間 23 ms / 5,000 ms
コード長 3,469 bytes
コンパイル時間 1,891 ms
コンパイル使用メモリ 159,656 KB
実行使用メモリ 5,204 KB
最終ジャッジ日時 2023-08-28 21:52:42
合計ジャッジ時間 2,997 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 23 ms
5,204 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 7 ms
4,968 KB
testcase_09 AC 7 ms
4,952 KB
testcase_10 AC 7 ms
4,620 KB
testcase_11 AC 2 ms
4,384 KB
testcase_12 AC 13 ms
4,376 KB
testcase_13 AC 18 ms
4,380 KB
testcase_14 AC 18 ms
4,376 KB
testcase_15 AC 1 ms
4,380 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: use of deprecated associated function `std::error::Error::description`: use the Display impl or to_string()
  --> Main.rs:96:62
   |
96 |             Err(why) => panic!("error in read_line: {}", why.description()),
   |                                                              ^^^^^^^^^^^
   |
   = note: `#[warn(deprecated)]` on by default

warning: 1 warning emitted

ソースコード

diff #

use std::io::{self, Stdin};
use std::str::{self, FromStr};
use std::error::Error;
use std::thread;
use std::cmp::max;
fn exec() {
    let mut sc = Scanner::new();
    let n: usize = sc.ne();
    let k: i32 = sc.ne();
    let w: Vec<i32> = (0..n).map(|_| sc.ne()).collect();
    let &ma = w.iter().max().unwrap();
    let &mi = w.iter().min().unwrap();
    let n = max(ma, mi * -1) as usize;
    let mut fwt = FenwickTree::new(n + 1, 0i32);
    for x in w {
        if x < 0 {
            let id = (x * -1) as usize;
            let num = fwt.sum(id + 1) - fwt.sum(id);
            if num > 0 {
                fwt.add(id, -1);
            }
        } else {
            let id = x as usize;
            let num = fwt.sum(n + 1) - fwt.sum(id);
            if num < k {
                fwt.add(id, 1);
            }
        }
    }
    println!("{}", fwt.sum(n + 1));
}
struct FenwickTree<T> {
    zero: T,
    n: usize,
    data: Vec<T>,
}
use std::ops::AddAssign;
impl<T: Copy + AddAssign> FenwickTree<T> {
    fn new(n_: usize, z_: T) -> FenwickTree<T> {
        FenwickTree {
            zero: z_,
            n: n_,
            data: vec![z_; n_],
        }
    }
    //sum of [0, j)
    fn sum(&self, j: usize) -> T {
        let mut res = self.zero;
        let mut x = j as i64 - 1;
        while x >= 0 {
            res += self.data[x as usize];
            x = (x & (x + 1)) - 1;
        }
        res
    }
    // add w to a[j]
    fn add(&mut self, j: usize, w: T) {
        let mut x = j;
        while x < self.n {
            self.data[x] += w;
            x |= x + 1;
        }
    }
}


fn main() {
    const DEFAULT_STACK: usize = 16 * 1024 * 1024;
    let builder = thread::Builder::new();
    let th = builder.stack_size(DEFAULT_STACK);
    let handle = th.spawn(|| { exec(); }).unwrap();
    let _ = handle.join();
}

#[allow(dead_code)]
struct Scanner {
    stdin: Stdin,
    id: usize,
    buf: Vec<u8>,
}

#[allow(dead_code)]
impl Scanner {
    fn new() -> Scanner {
        Scanner {
            stdin: io::stdin(),
            id: 0,
            buf: Vec::new(),
        }
    }
    fn next_line(&mut self) -> Option<String> {
        let mut res = String::new();
        match self.stdin.read_line(&mut res) {
            Ok(0) => return None,
            Ok(_) => Some(res),
            Err(why) => panic!("error in read_line: {}", why.description()),
        }
    }
    fn next<T: FromStr>(&mut self) -> Option<T> {
        while self.buf.len() == 0 {
            self.buf = match self.next_line() {
                Some(r) => {
                    self.id = 0;
                    r.trim().as_bytes().to_owned()
                }
                None => return None,
            };
        }
        let l = self.id;
        assert!(self.buf[l] != b' ');
        let n = self.buf.len();
        let mut r = l;
        while r < n && self.buf[r] != b' ' {
            r += 1;
        }
        let res = match str::from_utf8(&self.buf[l..r]).ok().unwrap().parse::<T>() {
            Ok(s) => Some(s),
            Err(_) => {
                panic!("parse error, {:?}",
                       String::from_utf8(self.buf[l..r].to_owned()))
            }
        };
        while r < n && self.buf[r] == b' ' {
            r += 1;
        }
        if r == n {
            self.buf.clear();
        } else {
            self.id = r;
        }
        res
    }
    fn ne<T: FromStr>(&mut self) -> T {
        self.next::<T>().unwrap()
    }
}
0