#177958 (Rust) No.515 典型LCP

提出ソース

結果

問題	No.515 典型LCP
ユーザー	tubo28
提出日時	2017-06-06 11:28:34
言語	Rust (1.77.0 + proconio)
結果	TLE
実行時間	-
コード長	12,324 bytes
コンパイル時間	13,922 ms
コンパイル使用メモリ	395,712 KB
実行使用メモリ	484,020 KB
最終ジャッジ日時	2024-09-22 11:41:28
合計ジャッジ時間	23,243 ms
ジャッジサーバーID （参考情報）	judge3 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	TLE	-
testcase_01	RE	-
testcase_02	RE	-
testcase_03	AC	1 ms 6,944 KB
testcase_04	AC	1 ms 6,944 KB
testcase_05	AC	907 ms 480,144 KB
testcase_06	AC	869 ms 479,568 KB
testcase_07	AC	878 ms 479,944 KB
testcase_08	AC	886 ms 479,136 KB
testcase_09	RE	-
testcase_10	RE	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	RE	-
testcase_15	AC	879 ms 484,020 KB
testcase_16	AC	874 ms 480,184 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

warning: field `n` is never read
   --> src/main.rs:339:5
    |
338 | struct SparceTable<T> {
    |        ----------- field in this struct
339 |     n: usize,
    |     ^
    |
    = note: `#[warn(dead_code)]` on by default

ソースコード

raw source code

// パフォーマスと使いやすさ再優先のためunsafeもunwrapもカジュアルに使う．
// 入力はASCII文字以外ダメ．
fn main() {
    let mut input_buffer = String::with_capacity(cp_io::BUFFER_SIZE);
    let mut output_buffer = Vec::with_capacity(cp_io::BUFFER_SIZE);
    unsafe {
        cp_io::input::buf = &mut output_buffer;
        cp_io::output::buf = &mut input_buffer;
    }
    std::thread::Builder::new()
        .stack_size(104_857_600)
        .spawn(solve).unwrap()
        .join().unwrap();
    cp_io::output::flush();
}

#[macro_use]
#[allow(dead_code, non_upper_case_globals, unused_macros)]
mod cp_io {
    pub const BUFFER_SIZE: usize = 8192;
    #[macro_use]
    pub mod output {
        pub const ENABLE_DUMP: bool = true;
        pub const PRECISION: usize = 10;
        pub static mut buf: *mut String = 0 as *mut String;

        macro_rules! p {
            ($x: expr) => {{
                $x.output();
                "\n".output();
                unsafe {
                    if (*cp_io::output::buf).len() > cp_io::BUFFER_SIZE {
                        flush();
                    }
                }
            }};
            ($x: expr, $($y: expr),+) => {{
                $x.output();
                " ".output();
                p!($($y),+);
            }};
        }

        macro_rules! pf {
            ($($x: expr),+) => {{
                p!($($x),+);
                flush();
            }};
        }

        macro_rules! d {
            ($($a: expr),+) => {
                if ENABLE_DUMP {
                    use std::io::*;
                    write!(stderr(), "{}:{}\t", file!(), line!()).unwrap();
                    d!(A $($a),+);
                    write!(stderr(), " = ").unwrap();
                    d!(B $($a),+);
                    write!(stderr(), "\n").unwrap();
                    stderr().flush().unwrap();
                }
            };
            (A $x: expr) => {
                write!(stderr(), "{}", stringify!($x)).unwrap();
            };
            (A $x: expr, $($y: expr),+) => {
                write!(stderr(), "{}, ", stringify!($x)).unwrap();
                d!(A $($y),+);
            };
            (B $x: expr) => {
                write!(stderr(), "{:?}", $x).unwrap();
            };
            (B $x: expr, $($y: expr),+) => {
                write!(stderr(), "{:?}, ", $x).unwrap();
                d!(B $($y),+);
            };
        }

        pub trait Output {
            fn output(&self);
        }

        macro_rules! output_normal {
            ($t: ty) => {
                impl Output for $t {
                    fn output(&self) {
                        unsafe {
                            use std::fmt::Write;
                            write!(&mut *buf, "{}", self).unwrap();
                        }
                    }
                }
            };
            ($t: ty, $($u: ty),+) => {
                output_normal!($t);
                output_normal!($($u),+);
            };
        }

        macro_rules! output_float {
            ($t: ty) => {
                impl Output for $t {
                    fn output(&self) {
                        unsafe {
                            use std::fmt::Write;
                            write!(&mut *buf, "{:.*}", PRECISION, self).unwrap();
                        }
                    }
                }
            };
            ($t: ty, $($u: ty),+) => {
                output_float!($t);
                output_float!($($u),+);
            };
        }

        pub fn flush() {
            unsafe {
                print!("{}", *buf);
                use std::io::*;
                stdout().flush().unwrap();
                (*buf).clear();
            }
        }

        output_normal!(u8, u16, u32, u64, usize);
        output_normal!(i8, i16, i32, i64, isize);
        output_normal!(bool, &'static str, String);
        output_float!(f32, f64);
    }

    pub mod input {
        pub trait Input<T> {
            fn input() -> T;
        }

        macro_rules! input_primitive {
            ($t: ty) => {
                impl Input<$t> for $t {
                    fn input() -> $t {
                        get_word().expect("EOF?").parse()
                            .unwrap_or_else(|e| panic!("Cannot parse {}", e))
                    }
                }
            };
            ($t: ty, $($u: ty),+) => {
                input_primitive!($t);
                input_primitive!($($u),+);
            };
        }

        macro_rules! input_tuple {
            ($($t: ident),*) => {
                impl< $($t: Input<$t>),* > Input< ( $($t),* ) > for ( $($t),* ) {
                    fn input() -> ( $($t),* ) {
                        ( $( $t::input()),* )
                    }
                }
            };
        }

        input_primitive!(u8, u16, u32, u64, usize);
        input_primitive!(i8, i16, i32, i64, isize);
        input_primitive!(bool, String);
        input_tuple!(A, B);
        input_tuple!(A, B, C);
        input_tuple!(A, B, C, D);

        pub fn get<T: Input<T>>() -> T {
            T::input()
        }

        pub fn get_vec<T: Input<T>>(n: usize) -> Vec<T> {
            (0..n).map(|_| get()).collect()
        }

        pub fn get_mat<T: Input<T>>(r: usize, c: usize) -> Vec<Vec<T>> {
            (0..r).map(|_| get_vec(c)).collect()
        }

        pub fn get_vec_char() -> Vec<char> {
            get_word().unwrap().chars().collect()
        }

        pub fn get_mat_char(h: usize) -> Vec<Vec<char>> {
            (0..h).map(|_| get_vec_char()).collect()
        }

        pub fn get_line() -> String {
            get_line_wrapped().unwrap()
        }

        fn get_word() -> Option<String> {
            let mut res = String::with_capacity(18);
            while let Some(c) = get_u8() {
                let d = c as char;
                if !d.is_whitespace() {
                    res.push(d);
                } else if res.len() != 0 {
                    unget_u8(c);
                    break;
                }
            }
            if res.len() == 0 { None } else { Some(res) }
        }

        fn get_line_wrapped() -> Option<String> {
            let c = get_u8();
            if c.is_none() {
                return None;
            }
            let mut line = String::with_capacity(18);
            line.push(c.unwrap() as char);
            loop {
                let c = get_u8();
                if c.is_none() || c.unwrap() == b'\n' {
                    // コメントはC++等での仕様
                    // if c.is_some() {
                    //     self.unget_u8(b'\n');
                    // }
                    return Some(line);
                }
                line.push(c.unwrap() as char);
            }
        }

        pub fn has_next() -> bool {
            loop {
                let c = get_u8();
                if c.is_none() {
                    return false;
                }
                let c = c.unwrap();
                if !(c as char).is_whitespace() {
                    unget_u8(c);
                    return true;
                }
            }
        }

        pub static mut buf: *mut Vec<u8> = 0 as *mut Vec<u8>;

        fn get_u8() -> Option<u8> {
            unsafe {
                let b = &mut (*buf);
                use std::io::{stdin, Read};
                if let Some(c) = b.pop() {
                    Some(c as u8)
                } else {
                    b.resize(super::BUFFER_SIZE, 0);
                    match stdin().read(b) {
                        Ok(l) if l > 0 => {
                            b.truncate(l);
                            b.reverse();
                            b.pop()
                        }
                        _ => return None,
                    }
                }
            }
        }

        fn unget_u8(c: u8) {
            unsafe {
                (*buf).push(c)
            }
        }
    }
}

#[allow(unused_imports)]
use cp_io::input::*;
#[allow(unused_imports)]
use cp_io::output::*;

use std::*;
use std::collections::*;
use std::mem::swap;
use std::hash::*;
use std::fmt::*;
use std::cmp::*;

struct Node<T> {
    depth: usize,
    next: BTreeMap<T, Box<Node<T>>>,
}

impl<T: Ord + Hash + Copy + Sized + Debug> Node<T> {
    fn new(depth: usize) -> Node<T> {
        Node {
            depth: depth,
            next: BTreeMap::new(),
        }
    }

    fn insert<'a, I>(&'a mut self, it: I) -> *const Node<T>
        where I: IntoIterator<Item=&'a T>
    {
        let mut it = it.into_iter();
        if let Some(c) = it.next() {
            let d = self.depth;
            self.next.entry(*c).or_insert_with(|| Box::new(Self::new(d + 1))).insert(it)
        } else {
            self
        }
    }
}

struct Trie<T> {
    root: Node<T>,
}

impl<T: Hash + Copy + Ord + Debug> Trie<T> {
    fn new() -> Self {
        Trie {
            root: Node::new(0),
        }
    }
    fn insert<'a, I>(&'a mut self, it: I) -> *const Node<T>
        where I: IntoIterator<Item=&'a T>
    {
        self.root.insert(it)
    }
    fn dfs(n: &Node<T>, a: &mut Vec<*const Node<T>>) {
        a.push(n);
        for (_, ch) in &n.next {
            Self::dfs(ch, a);
            a.push(n);
        }
    }
    fn make_array(&self) -> Vec<*const Node<T>> {
        let mut a = vec![];
        Trie::dfs(&self.root, &mut a);
        a
    }
}

struct SparceTable<T> {
    n: usize,
    min: Vec<Vec<T>>
}

impl<T : Copy + Ord + Debug> SparceTable<T> {
    fn new<I>(it: I) -> SparceTable<T>
        where I: IntoIterator<Item=T> {
        let it = it.into_iter();
        let v = it.collect::<Vec<_>>();
        let n = v.len();
        let mut res = SparceTable {
            n: n,
            min: vec![v],
        };
        for i in 0..30 {
            let w = 1 << i;
            if n < w*2-1 {
                break;
            }
            let next = (0..n-w*2-1).map(|l| {
                min(res.min[i][l], res.min[i][l + w])
            }).collect();
            // d!(next);
            res.min.push(next);
        }
        res
    }

    fn rmq(&self, l: usize, r: usize) -> T {
        let w = r - l;
        // d!(l, r, w);
        let mut i = 0;
        while (1 << i) < w {
            i += 1;
        }
        i -= 1;
        let rr = r - (1 << i);
        //assert!(r - (1 << i) == rr);
        min(self.min[i][l], self.min[i][rr])
    }
}

fn solve() {
    while has_next() {
        let n = get();
        let s: Vec<String> = get_vec(n);

        let (m, mut x, d): (usize, u64, u64) = get();
        let mut i = vec![0usize; m];
        let mut j = vec![0usize; m];
        for k in 0..m {
            let n = n as u64;
            i[k] = ((x / (n - 1)) + 1) as usize;
            j[k] = ((x % (n - 1)) + 1) as usize;
            if i[k] > j[k] {
                swap(&mut i[k], &mut j[k]);
            } else {
                j[k] = j[k] + 1;
            }
            x = (x + d) % (n * (n - 1));
        }

        let mut trie = Trie::<char>::new();
        let (index, depth) = {
            let s: Vec<Vec<char>> = s.iter().map(|s| s.chars().collect()).collect();
            let node = s.iter().map(|s| trie.insert(s.iter())).collect::<Vec<_>>();
            let pa = trie.make_array();
            let mut mp = BTreeMap::<_, usize>::new();
            for i in 0..pa.len() {
                mp.entry(pa[i]).or_insert(i);
            }
            let index = (0..n).map(|i| mp[&node[i]]).collect::<Vec<usize>>();
            let da = pa.iter().map(|i| unsafe { (**i).depth }).collect::<Vec<usize>>();
            (index, da)
        };

        let rmq = SparceTable::<_>::new(depth.iter());
        // d!(rmq.min);

        // d!(index, depth);
        let calc = |i: usize, j: usize| {
            let (l, r) = if index[i] < index[j] {
                (index[i], index[j])
            } else {
                (index[j], index[i])
            };
            rmq.rmq(l, r + 1)
            // (l..r+1).fold(1000000000, |acc, i| if acc > depth[i] { depth[i] } else { acc } )
        };
        let ans = i.iter().zip(j.iter()).fold(0, |acc, (&i, &j)| {
            let res = calc(i - 1, j - 1);
            // d!(res);
            acc + res
        });
        p!(ans);
    }
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード