結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー cympfh
提出日時 2018-12-27 14:44:13
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 2,691 bytes
コンパイル時間 11,364 ms
コンパイル使用メモリ 406,204 KB
実行使用メモリ 41,224 KB
最終ジャッジ日時 2024-10-01 14:52:00
合計ジャッジ時間 12,996 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#[allow(unused_macros)]
macro_rules! trace {
    ($var:expr) => {
        let _ = writeln!(&mut std::io::stderr(), ">>> {} = {:?}", stringify!($var), $var);
    };
    ($($args:expr),*) => { trace!(($($args),*)) }
}

#[allow(unused_macros)]
macro_rules! put {
    ($var:expr) => {
        let _ = writeln!(&mut std::io::stdout(), "{}", $var);
    };
    ($var:expr, $($args:expr),*) => {
        let _ = write!(&mut std::io::stdout(), "{} ", $var);
        put!($($args),*);
    };
}

use std::ops::Range;

struct BIT { size: usize, array: Vec<i32> }

#[allow(dead_code)]
impl BIT {
    fn new(n: usize) -> BIT {
        BIT  { size: n, array: vec![0; n + 1] }
    }
    fn add(&mut self, idx: usize, w: i32) {
        let mut x = idx + 1;
        while x <= self.size {
            self.array[x] += w;
            let xi = x as i32;
            x += (xi&-xi) as usize;
        }
    }
    fn sum_up(&self, idx: usize) -> i32 { // [0, idx)
        let mut sum = 0;
        let mut x = idx;
        while x > 0 {
            sum += self.array[x];
            let xi = x as i32;
            x -= (xi&-xi) as usize;
        }
        sum
    }
    fn sum(&self, range: Range<usize>) -> i32 {
        self.sum_up(range.end) - self.sum_up(range.start)
    }
}

fn bfs(u: usize, mut bit: &mut BIT, tree: &Vec<Vec<usize>>) -> i32 {
    let mut a = bit.sum_up(u);
    bit.add(u, 1);
    for &v in tree[u].iter() {
        a += bfs(v, &mut bit, &tree);
    }
    bit.add(u, -1);
    a
}

fn main() {
    let mut sc = Scanner::new();
    let n: usize = sc.cin();
    let mut bit = BIT::new(n);
    let mut tree = vec![vec![]; n];
    // let mut rev = vec![None; n];
    for i in 1..n {
        let a: usize = sc.cin();
        tree[a].push(i);
        // rev[i] = Some(a);
    }

    let ans = bfs(0, &mut bit, &tree);
    put!(ans);
}

#[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")
        }
    }
}
0