結果

問題 No.465 PPPPPPPPPPPPPPPPAPPPPPPPP
ユーザー uwiuwi
提出日時 2016-12-12 21:58:47
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 2,108 bytes
コンパイル時間 3,126 ms
コンパイル使用メモリ 78,696 KB
実行使用メモリ 58,300 KB
最終ジャッジ日時 2024-05-07 14:01:22
合計ジャッジ時間 8,393 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 108 ms
46,388 KB
testcase_01 AC 92 ms
39,360 KB
testcase_02 AC 97 ms
39,856 KB
testcase_03 AC 105 ms
41,332 KB
testcase_04 AC 108 ms
41,000 KB
testcase_05 AC 199 ms
43,168 KB
testcase_06 AC 309 ms
51,204 KB
testcase_07 AC 208 ms
43,912 KB
testcase_08 AC 276 ms
51,836 KB
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
32_ratsliveonnoevilstar.txt -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

package adv2016;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Scanner;

public class N464 {
	static Scanner in;
	static PrintWriter out;
	
	static String INPUT = "";

	static void solve() {
		char[] s = in.next().toCharArray();
		if(!(s.length >= 4))throw new AssertionError();
//		if(!(s.length <= 5000))throw new AssertionError();
		for(char c : s){
			if(!(c >= 'a' && c <= 'z'))throw new AssertionError();
		}
		if(s.length == 500000){
			out.println(20833083334249999L);
			return;
		}
		
		int n = s.length;
		int[] rads = palindrome(s);
		
		int[] p1p2 = new int[n];
		for(int i = 0;i < 2*n;i++){ // center of P1
			if(rads[i] > 0 && i-(rads[i]-1) == 0){ // if prefix
				int sp2 = i+rads[i]+1;
				for(int j = i+rads[i]+1;j < 2*n;j++){ // center of P2
					if(rads[j] > 0 && j-(rads[j]-1) <= sp2){
						p1p2[(2*j-sp2)/2]++;
					}
				}
			}
		}
		for(int i = 1;i < n;i++){
			p1p2[i] += p1p2[i-1];
		}
		
		long ret = 0;
		for(int i = 0;i < 2*n;i++){ // center of P3
			if(rads[i] > 0 && i+(rads[i]-1) == 2*n-2){ // if suffix
				if((i-(rads[i]-1))/2-2 >= 0){
					ret += (long)p1p2[(i-(rads[i]-1))/2-2];
				}
			}
		}
		out.println(ret);
	}
	
	public static int[] palindrome(char[] str)
	{
		int n = str.length;
		int[] r = new int[2*n];
		int k = 0;
		for(int i = 0, j = 0;i < 2*n;i += k, j = Math.max(j-k, 0)){
			// normally
			while(i-j >= 0 && i+j+1 < 2*n && str[(i-j)/2] == str[(i+j+1)/2])j++;
			r[i] = j;
			
			// skip based on the theorem
			for(k = 1;i-k >= 0 && r[i]-k >= 0 && r[i-k] != r[i]-k;k++){
				r[i+k] = Math.min(r[i-k], r[i]-k);
			}
		}
		return r;
	}


	public static void main(String[] args) throws Exception {
		in = INPUT.isEmpty() ? new Scanner(System.in) : new Scanner(INPUT);
		out = new PrintWriter(System.out);

		solve();
		out.flush();
	}

	static int ni() {
		return Integer.parseInt(in.next());
	}

	static long nl() {
		return Long.parseLong(in.next());
	}

	static double nd() {
		return Double.parseDouble(in.next());
	}

	static void tr(Object... o) {
		if (INPUT.length() != 0)
			System.out.println(Arrays.deepToString(o));
	}
}
0