結果

問題 No.464 PPAP
ユーザー hatoohatoo
提出日時 2017-11-06 15:58:41
言語 Rust
(1.77.0)
結果
AC  
実行時間 406 ms / 2,000 ms
コード長 4,220 bytes
コンパイル時間 13,012 ms
コンパイル使用メモリ 402,008 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-03 06:20:26
合計ジャッジ時間 14,772 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 12 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 406 ms
5,376 KB
testcase_11 AC 19 ms
5,376 KB
testcase_12 AC 36 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 1 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 1 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 1 ms
5,376 KB
testcase_25 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[allow(unused_imports)]
use std::cmp::{max, min, Ordering};
#[allow(unused_imports)]
use std::collections::{HashMap, HashSet, BinaryHeap, VecDeque};
#[allow(unused_imports)]
use std::iter::FromIterator;

mod util {
    use std::io::stdin;
    use std::str::FromStr;
    use std::fmt::Debug;

    #[allow(dead_code)]
    pub fn line() -> String {
        let mut line: String = String::new();
        stdin().read_line(&mut line).unwrap();
        line.trim().to_string()
    }

    #[allow(dead_code)]
    pub fn get<T: FromStr>() -> T
    where
        <T as FromStr>::Err: Debug,
    {
        let mut line: String = String::new();
        stdin().read_line(&mut line).unwrap();
        line.trim().parse().unwrap()
    }

    #[allow(dead_code)]
    pub fn gets<T: FromStr>() -> Vec<T>
    where
        <T as FromStr>::Err: Debug,
    {
        let mut line: String = String::new();
        stdin().read_line(&mut line).unwrap();
        line.split_whitespace()
            .map(|t| t.parse().unwrap())
            .collect()
    }

    #[allow(dead_code)]
    pub fn get2<T: FromStr, U: FromStr>() -> (T, U)
    where
        <T as FromStr>::Err: Debug,
        <U as FromStr>::Err: Debug,
    {
        let mut line: String = String::new();
        stdin().read_line(&mut line).unwrap();
        let mut iter = line.split_whitespace();
        (
            iter.next().unwrap().parse().unwrap(),
            iter.next().unwrap().parse().unwrap(),
        )
    }

    #[allow(dead_code)]
    pub fn get3<S: FromStr, T: FromStr, U: FromStr>() -> (S, T, U)
    where
        <S as FromStr>::Err: Debug,
        <T as FromStr>::Err: Debug,
        <U as FromStr>::Err: Debug,
    {
        let mut line: String = String::new();
        stdin().read_line(&mut line).unwrap();
        let mut iter = line.split_whitespace();
        (
            iter.next().unwrap().parse().unwrap(),
            iter.next().unwrap().parse().unwrap(),
            iter.next().unwrap().parse().unwrap(),
        )
    }
}

#[allow(unused_macros)]
macro_rules! debug {
    ($x: expr) => {
        println!("{}: {:?}", stringify!($x), $x)
    }
}

struct Manacher {
    r: Vec<usize>,
}

impl Manacher {
    fn new<T: Eq>(seq: &[T]) -> Manacher {
        let mut r = vec![0; 2 * seq.len() - 1];
        let mut i = 0;
        let mut j = 0;
        while i < r.len() {
            while i >= j && i + j < r.len() && Self::get(seq, i - j) == Self::get(seq, i + j) {
                j += 1;
            }
            r[i] = j;
            let mut k = 1;
            while i >= k && i + k < r.len() && k + r[i - k] < j {
                r[i + k] = r[i - k];
                k += 1;
            }
            i += k;
            j -= k;
        }

        Manacher { r: r }
    }

    // [l,r]
    fn is_palindrome(&self, l: usize, r: usize) -> bool {
        self.r[l + r] >= r - l + 1
    }

    fn get<T: Eq>(s: &[T], i: usize) -> Option<&T> {
        if i & 1 == 1 { None } else { s.get(i >> 1) }
    }
}

fn main() {
    let s: Vec<char> = util::line().chars().collect();
    // let s_rev: Vec<char> = s.iter().cloned().rev().collect();
    let mut front = Vec::new();
    let mut back = Vec::new();

    {
        let mut a = VecDeque::new();
        let mut b = VecDeque::new();

        for (i, &c) in s.iter().enumerate() {
            a.push_front(c);
            b.push_back(c);

            if a == b {
                front.push(i);
            }
        }
    }

    {
        let mut a = VecDeque::new();
        let mut b = VecDeque::new();

        for (i, &c) in s.iter().enumerate().rev() {
            a.push_front(c);
            b.push_back(c);

            if a == b {
                back.push(i);
            }
        }
    }

    let mut ans = 0;
    back.sort();
    let manacher = Manacher::new(&s);
    for &p in &front {
        for (_, i) in s[p + 1..].iter().zip(1..) {
            if manacher.is_palindrome(p + 1, p + i) {
                ans += back.len() as u64 -
                    match back.binary_search(&(p + i as usize + 2)) {
                        Ok(x) => x as u64,
                        Err(x) => x as u64,
                    };
            }
        }

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

}
0