結果

問題 No.464 PPAP
コンテスト
ユーザー hatoo
提出日時 2017-11-06 15:58:41
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 402 ms / 2,000 ms
コード長 4,220 bytes
コンパイル時間 13,870 ms
コンパイル使用メモリ 377,520 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-24 04:03:20
合計ジャッジ時間 13,923 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

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