結果

問題 No.263 Common Palindromes Extra
ユーザー tanzakutanzaku
提出日時 2016-12-18 03:10:17
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 10,108 bytes
コンパイル時間 6,424 ms
コンパイル使用メモリ 86,100 KB
実行使用メモリ 124,440 KB
最終ジャッジ日時 2024-05-08 06:15:26
合計ジャッジ時間 6,935 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 76 ms
55,820 KB
testcase_01 AC 54 ms
54,900 KB
testcase_02 AC 54 ms
54,496 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 168 ms
60,404 KB
testcase_06 AC 72 ms
56,068 KB
testcase_07 WA -
testcase_08 AC 336 ms
99,008 KB
testcase_09 WA -
testcase_10 AC 435 ms
122,296 KB
testcase_11 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.*;
import java.math.*;
import java.util.*;

import static java.util.Arrays.*;

public class _263 {
	private static final int mod = (int)1e9+7;

	final Random random = new Random(0);
	final IOFast io = new IOFast();

	/// MAIN CODE
	public void run() throws IOException {
//		int TEST_CASE = Integer.parseInt(new String(io.nextLine()).trim());
		int TEST_CASE = 1;
		while(TEST_CASE-- != 0) {
			char[] S = io.next();
			char[] T = io.next();

			PalindromicTree t1 = new PalindromicTree(S);
			PalindromicTree t2 = new PalindromicTree(T);
			
//			dump(PalindromicTree.cnt);
//			if (true) return;
			t1.prepare();
			t2.prepare();

			long ans = 0;
			ans += PalindromicTree.dfs(t1.rootEven, t2.rootEven);
			ans += PalindromicTree.dfs(t1.rootOdd, t2.rootOdd);
			io.out.println(ans);
		}
	}
	
	static
	public class PalindromicTree {
		
		static class Node {
//			public Node(int len, String s) {
			public Node(int len, int ch) {
				this.len = len;
//				this.s = s;
				this.ans = -1;
				this.ch = (byte)ch;
			}
			
			long ans;
			int cnt;
			int len;
			int in;
			byte ch;
			Node suffixLink;
			Node head;
			Node sibling;
			
//			String s;
			
//			@Override
//			public String toString() {
//				// TODO Auto-generated method stub
//				String res = "str:[" + s + "]";
//				return res;
//			}
		}

		void prepare() {
			List<Node> xs = new ArrayList<>();
			prepare0(rootEven, xs);
			prepare0(rootOdd, xs);
			prepare1(xs);
			
//			prepare(rootEven);
//			prepare(rootOdd);
		}

		void prepare0(Node n, List<Node> xs) {
			int t = 0;
			st[0][t++] = n;
			while (t > 0) {
				n = st[0][--t];
				xs.add(n);
				n.suffixLink.in++;
				for (Node x = n.head; x != null; x = x.sibling) {
					st[0][t++] = x;
				}
			}
		}

		void prepare1(List<Node> xs) {
			Node[] q = new Node[xs.size()];
			int s = 0, t = 0;
			for (Node n : xs) {
//				dump("in", n.s, n.in);
				if (n.in == 0) q[t++] = n;
			}
			while (s != t) {
				Node n = q[s++];
//				n.suffixLink.cnt += n.suffixLink.len <= n.len/2 ? 1 : 2;
				n.suffixLink.cnt += n.cnt;
//				dump(n.s, n.suffixLink.s);
				if (--n.suffixLink.in == 0) {
					q[t++] = n.suffixLink;
				}
			}
		}

		long prepare(Node n) {
			for (Node x = n.head; x != null; x = x.sibling) {
				prepare(x);
			}
//			dump(s, n, n.cnt);
			return n.cnt;
		}

//		static long dfs(Node n1, Node n2) {
//			if (n1 == null || n2 == null) return 0;
//			long ans = 0;
//			if (n1.len > 0) ans += n1.cnt * n2.cnt;
////			dump(n1.s, n2.s, n1.cnt, n2.cnt);
//			if (n1.next != null || n2.next != null) {
//				for (int i = 0; i < 26; i++) {
//					ans += dfs(n1.next[i], n2.next[i]);
//				}
//			}
//			return ans;
//		}

		static Node[][] st = new Node[2][500000+100];
		static long dfs(Node n1, Node n2) {
			int t = 0;
			st[0][t] = n1;
			st[1][t] = n2;
			t++;
			while (t > 0) {
				Node n3 = st[0][--t];
				Node n4 = st[1][t];
				if (n3 == null || n4 == null) continue;
				if (n3.ans == -1) {
					n3.ans = -2;
					st[0][t] = n3;
					st[1][t] = n4;
					t++;
					for (Node x = n3.head; x != null; x = x.sibling) {
						for (Node y = n4.head; y != null; y = y.sibling) {
							if (x.ch == y.ch) {
								st[0][t] = x;
								st[1][t] = y;
								t++;
								break;
							}
						}
					}
				} else if (n3.ans == -2) {
					long v = 0;
					if (n3.len > 0) v = n3.cnt * n4.cnt;
					for (Node x = n3.head; x != null; x = x.sibling) {
						for (Node y = n4.head; y != null; y = y.sibling) {
							if (x.ch == y.ch) {
								v += x.ans;
							}
						}
					}
					n3.ans = v;
				}
			}
			return n1.ans;
		}
		
		char[] cs;
		private Node next(Node n, int i) {
//			dump(1, n, n.len);
			while (true) {
				final int l = i - (n.len + 1);
//				dump("search: ", n, l, i);
				if (l >= 0 && cs[l] == cs[i]) {
//					dump("find: ", n, l, i);
					break;
				}
//				dump(i, l, n.len);
				if (n == n.suffixLink) throw new RuntimeException();
				n = n.suffixLink;
			}
			int to = cs[i] - 'A';
			Node next = null;
			for (Node x = n.head; x != null; x = x.sibling) {
				if (x.ch == to) {
					next = x;
					break;
				}
			}
			if (next == null) {
//				n.next[to] = new Node(n.len + 2, cs[i] + n.s + cs[i]);
				next = new Node(n.len + 2, to);
//				dump(2, n.len, next(n.suffixLink, i).len);
				next.sibling = n.head;
				n.head = next;
				next.suffixLink = next(n.suffixLink, i);
				if (next.suffixLink == next) {
					next.suffixLink= rootEven;
				}
			}
			return next;
		}
		
		final int ALPHABET_SIZE = 26;

		Node rootOdd;
		Node rootEven;
		public PalindromicTree(char[] cs) {
			this.cs = cs;
			
			// 長さ-1,0のノードを作成
//			rootEven = new Node(0, "");
//			rootOdd = new Node(-1, "$");
			rootEven = new Node(0, -1);
			rootOdd = new Node(-1, -2);
			rootEven.suffixLink = rootOdd;
			rootOdd.suffixLink = rootOdd;
			
			Node cur = rootEven;
			
			for (int i = 0; i < cs.length; i++) {
//				dump("func", i);
				cur = next(cur, i);
				cur.cnt++;
			}
		}
	}
	
	/// TEMPLATE
	static int gcd(int n, int r) { return r == 0 ? n : gcd(r, n%r); }
	static long gcd(long n, long r) { return r == 0 ? n : gcd(r, n%r); }
	
	static <T> void swap(T[] x, int i, int j) { T t = x[i]; x[i] = x[j]; x[j] = t; }
	static void swap(int[] x, int i, int j) { int t = x[i]; x[i] = x[j]; x[j] = t; }

	void printArrayLn(int[] xs) { for(int i = 0; i < xs.length; i++) io.out.print(xs[i] + (i==xs.length-1?"\n":" ")); }
	void printArrayLn(long[] xs) { for(int i = 0; i < xs.length; i++) io.out.print(xs[i] + (i==xs.length-1?"\n":" ")); }
	
	static void dump(Object... o) { System.err.println(Arrays.deepToString(o)); } 
	
	void main() throws IOException {
		//		IOFast.setFileIO("rle-size.in", "rle-size.out");
		try { run(); }
		catch (EndOfFileRuntimeException e) { }
		io.out.flush();
	}
	public static void main(String[] args) throws IOException { new _263().main(); }
	
	static class EndOfFileRuntimeException extends RuntimeException {
		private static final long serialVersionUID = -8565341110209207657L; }

	static
	public class IOFast {
		private InputStream in = System.in;
		private PrintWriter out = new PrintWriter(System.out);

		private static int pos, readLen;
		private static final byte[] buffer = new byte[1024 * 8];
		private static char[] str = new char[500*8*2];
		private static boolean[] isDigit = new boolean[256];
		private static boolean[] isSpace = new boolean[256];
		private static boolean[] isLineSep = new boolean[256];

		static { for(int i = 0; i < 10; i++) { isDigit['0' + i] = true; } isDigit['-'] = true; isSpace[' '] = isSpace['\r'] = isSpace['\n'] = isSpace['\t'] = true; isLineSep['\r'] = isLineSep['\n'] = true; }
		public byte read() throws IOException { if(pos >= readLen) { pos = 0; readLen = in.read(buffer); if(readLen <= 0) { throw new EndOfFileRuntimeException(); } } return buffer[pos++]; }
		public int nextInt() throws IOException {
			byte first = (byte)nextChar();
			boolean minus = false;
			int res = 0;
			if (first == '-') {
				minus = true;
			} else {
				res = first - '0';
			}
			while (true) {
				byte b = read();
				if (isSpace[b]) break;
				res = res * 10 + b - '0';
			}
			return minus ? -res : res;
		}
		public long nextLong() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isSpace); int i = 0; long ret = 0; if(str[0] == '-') { i = 1; } for(; i < len; i++) ret = ret * 10 + str[i] - '0'; if(str[0] == '-') { ret = -ret; } return ret; }
		public char nextChar() throws IOException { while(true) { final int c = read(); if(!isSpace[c]) { return (char)c; } } }
		int reads(int len, boolean[] accept) throws IOException { try { while(true) { final int c = read(); if(accept[c]) { break; } if(str.length == len) { char[] rep = new char[str.length * 3 / 2]; System.arraycopy(str, 0, rep, 0, str.length); str = rep; } str[len++] = (char)c; } } catch(EndOfFileRuntimeException e) { ; } return len; }
		int reads(char[] cs, int len, boolean[] accept) throws IOException { try { while(true) { final int c = read(); if(accept[c]) { break; } cs[len++] = (char)c; } } catch(EndOfFileRuntimeException e) { ; } return len; }
		public char[] nextLine() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isLineSep); try { if(str[len-1] == '\r') { len--; read(); } } catch(EndOfFileRuntimeException e) { ; } return Arrays.copyOf(str, len); }
		public String nextString() throws IOException { return new String(next()); }
		public char[] next() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isSpace); return Arrays.copyOf(str, len); }
//		public int next(char[] cs) throws IOException { int len = 0; cs[len++] = nextChar(); len = reads(cs, len, isSpace); return len; }
//		public double nextDouble() throws IOException { return Double.parseDouble(nextString()); }
		public double nextDouble() throws IOException {
			double res = nextChar() - '0';
			while (true) {
				byte c = read();
				if (isSpace[c]) return res;
				if (c == '.') break;
				res = res * 10 + c - '0';
			}
			double d = 1;
			while (true) {
				byte c = read();
//				dump(c);
				if (!isDigit[c]) break;
				res = res * 10 + c - '0';
				d *= 10;
			}
//			dump(1);
			return res / d;
		}
		public long[] nextLongArray(final int n) throws IOException { final long[] res = new long[n]; for(int i = 0; i < n; i++) { res[i] = nextLong(); } return res; }
		public int[] nextIntArray(final int n) throws IOException { final int[] res = new int[n]; for(int i = 0; i < n; i++) { res[i] = nextInt(); } return res; }
		public int[][] nextIntArray2D(final int n, final int k) throws IOException { final int[][] res = new int[n][]; for(int i = 0; i < n; i++) { res[i] = nextIntArray(k); } return res; }
		public int[][] nextIntArray2DWithIndex(final int n, final int k) throws IOException { final int[][] res = new int[n][k+1]; for(int i = 0; i < n; i++) { for(int j = 0; j < k; j++) { res[i][j] = nextInt(); } res[i][k] = i; } return res; }
		public double[] nextDoubleArray(final int n) throws IOException { final double[] res = new double[n]; for(int i = 0; i < n; i++) { res[i] = nextDouble(); } return res; }
	}
}
0