#153020 (Rust) No.489 株に挑戦

• 問題: No.489 株に挑戦
• ユーザー: cympfh
• 提出日時: 2017-02-26 00:13:43
• 言語: Rust (1.77.0)
• 結果: AC
• 実行時間: 50 ms / 1,000 ms
• コード長: 4,070 bytes
• コンパイル時間: 1,528 ms
• コンパイル使用メモリ: 153,372 KB
• 実行使用メモリ: 8,540 KB
• 最終ジャッジ日時: 2023-09-27 05:26:52
• 合計ジャッジ時間: 3,336 ms
• ジャッジサーバーID （参考情報）: judge13 / judge11

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	28 ms; 8,056 KB
testcase_01	AC	25 ms; 5,152 KB
testcase_02	AC	1 ms; 4,380 KB
testcase_03	AC	1 ms; 4,376 KB
testcase_04	AC	1 ms; 4,376 KB
testcase_05	AC	1 ms; 4,376 KB
testcase_06	AC	1 ms; 4,376 KB
testcase_07	AC	1 ms; 4,376 KB
testcase_08	AC	1 ms; 4,376 KB
testcase_09	AC	1 ms; 4,376 KB
testcase_10	AC	1 ms; 4,380 KB
testcase_11	AC	1 ms; 4,380 KB
testcase_12	AC	1 ms; 4,380 KB
testcase_13	AC	1 ms; 4,380 KB
testcase_14	AC	1 ms; 4,376 KB
testcase_15	AC	2 ms; 4,380 KB
testcase_16	AC	43 ms; 8,408 KB
testcase_17	AC	3 ms; 4,376 KB
testcase_18	AC	20 ms; 5,188 KB
testcase_19	AC	20 ms; 5,116 KB
testcase_20	AC	46 ms; 8,404 KB
testcase_21	AC	11 ms; 4,376 KB
testcase_22	AC	3 ms; 4,380 KB
testcase_23	AC	14 ms; 5,224 KB
testcase_24	AC	50 ms; 8,480 KB
testcase_25	AC	1 ms; 4,380 KB
testcase_26	AC	49 ms; 8,540 KB
testcase_27	AC	40 ms; 8,532 KB
testcase_28	AC	1 ms; 4,380 KB
testcase_29	AC	1 ms; 4,376 KB
testcase_30	AC	48 ms; 8,536 KB
testcase_31	AC	48 ms; 8,520 KB
testcase_32	AC	23 ms; 5,256 KB
testcase_33	AC	49 ms; 8,444 KB
testcase_34	AC	40 ms; 8,384 KB
testcase_35	AC	17 ms; 4,768 KB
testcase_36	AC	6 ms; 4,380 KB
testcase_37	AC	25 ms; 5,244 KB

コンパイルメッセージ

warning: unused macro definition: `trace`
 --> Main.rs:7:14
  |
7 | macro_rules! trace {
  |              ^^^^^
  |
  = note: `#[warn(unused_macros)]` on by default

warning: unused macro definition: `swap`
  --> Main.rs:12:14
   |
12 | macro_rules! swap { ($a:expr, $b:expr) => ({ let t = $b; $b = $a; $a = t; }) }
   |              ^^^^

warning: variable `k` is assigned to, but never used
  --> Main.rs:17:32
   |
17 |         let mut m = 1; let mut k = 1;
   |                                ^
   |
   = note: consider using `_k` instead
   = note: `#[warn(unused_variables)]` on by default

warning: associated function `update` is never used
  --> Main.rs:60:8
   |
60 |     fn update(&mut self, idx: usize, x: i64) {
   |        ^^^^^^
   |
   = note: `#[warn(dead_code)]` on by default

warning: associated function `add` is never used
  --> Main.rs:71:8
   |
71 |     fn add(&mut self, idx: usize, x: i64) {
   |        ^^^

warning: associated function `at` is never used
  --> Main.rs:76:8
   |
76 |     fn at(&self, idx: usize) -> i64 {
   |        ^^

warning: 6 warnings emitted

ソースコード

#![allow(unused_imports)]
use std::io::{ self, Write };
use std::str::FromStr;
use std::cmp::{ min, max };
use std::collections::{ BinaryHeap, VecDeque };

macro_rules! trace {
    ($var:expr) => ({
        let _ = writeln!(&mut std::io::stderr(), ">>> {} = {:?}", stringify!($var), $var);
    })
}
macro_rules! swap { ($a:expr, $b:expr) => ({ let t = $b; $b = $a; $a = t; }) }
struct RMQ {heap: Vec<i64>, size: usize, lefts: Vec<usize>, rights: Vec<usize>}
impl RMQ {

    fn alignment(n: usize) -> usize {
        let mut m = 1; let mut k = 1;
        while n > m { k += 1; m *= 2; }
        m * 2 - 1
    }

    fn of_left(idx: usize, base: usize) -> usize {
        let mut i = idx; while i < base { i = i * 2 + 1; }
        i - base
    }

    fn of_right(idx: usize, base: usize) -> usize {
        let mut i = idx; while i < base { i = i * 2 + 2; }
        i - base
    }

    fn new(v: &Vec<i64>) -> RMQ {
        let n = v.len();
        let size = RMQ::alignment(n);
        let mut heap = vec![0; size];
        for i in 0..n { heap[i + size/2] = v[i]; }
        for i in (0..size/2).rev() { heap[i] = max(heap[i*2+1], heap[i*2+2]); }
        let lefts = (0..size).map(|i| RMQ::of_left(i, size/2)).collect();
        let rights = (0..size).map(|i| RMQ::of_right(i, size/2)).collect();
        RMQ {heap: heap, size: size, lefts: lefts, rights: rights}
    }

    // max [left..right] inclusively
    fn max(&self, left: usize, right: usize) -> i64 {
        if left > right { return 0 }
        if left == right { return self.heap[self.size/2 + left] }
        let mut i = self.size/2 + left;
        loop {
            let parent = (i - 1) / 2;
            if self.lefts[parent] == left && self.rights[parent] <= right {
                i = parent;
            } else {
                break
            }
        }
        max(self.heap[i], self.max(self.rights[i] + 1, right))
    }

    // [idx] <- x
    fn update(&mut self, idx: usize, x: i64) {
        let mut i = idx + self.size/2;
        self.heap[i] = x;
        loop {
            i = (i - 1) / 2;
            if i == 0 { break }
            self.heap[i] = max(self.heap[i*2+1], self.heap[i*2+2]);
        }
    }

    // [idx] <- [idx] + x
    fn add(&mut self, idx: usize, x: i64) {
        let z = self.heap[idx + self.size/2] + x;
        self.update(idx, z);
    }

    fn at(&self, idx: usize) -> i64 {
        self.heap[idx + self.size/2]
    }
}

fn main() {
    let mut sc = Scanner::new();
    let n: usize = sc.cin();
    let d: usize = sc.cin();
    let k: i64 = sc.cin();
    let xs = (0..n).map(|_| sc.cin()).collect::<Vec<i64>>();

    let rmq = RMQ::new(&xs);

    let mut ans = 0;
    let mut a = 0;

    for i in 0..(n-1) {
        let mx = rmq.max(i + 1, min(i + d, n - 1));
        let p = mx - xs[i];
        if ans < p {
            ans = p;
            a = i;
        }
    }

    println!("{}", ans * k);

    if ans > 0 {
        for j in (a+1)..n {
            if ans == xs[j] - xs[a] {
                println!("{} {}", a, j);
                break;
            }
        }
    }
}

#[allow(dead_code)]
struct Scanner { stdin: io::Stdin, buffer: VecDeque<String>, }
#[allow(dead_code)]
impl Scanner {
    fn new() -> Scanner { Scanner { stdin: io::stdin(), buffer: VecDeque::new() } }
    fn reserve(&mut self) {
        while self.buffer.len() == 0 {
            let mut line = String::new();
            let _ = self.stdin.read_line(&mut line);
            for w in line.split_whitespace() {
                self.buffer.push_back(String::from(w));
            }
        }
    }
    fn cin<T: FromStr>(&mut self) -> T {
        self.reserve();
        match self.buffer.pop_front().unwrap().parse::<T>() {
            Ok(a) => a,
            Err(_) => panic!("parse err")
        }
    }
    fn get_char(&mut self) -> char {
        self.reserve();
        let head = self.buffer[0].chars().nth(0).unwrap();
        let tail = String::from( &self.buffer[0][1..] );
        if tail.len()>0 { self.buffer[0]=tail } else { self.buffer.pop_front(); }
        head
    }
}