結果

問題 No.430 文字列検索
ユーザー 37zigen37zigen
提出日時 2016-10-28 15:14:30
言語 Java21
(openjdk 21)
結果
AC  
実行時間 232 ms / 2,000 ms
コード長 2,240 bytes
コンパイル時間 3,440 ms
コンパイル使用メモリ 79,916 KB
実行使用メモリ 47,052 KB
最終ジャッジ日時 2024-11-10 00:12:11
合計ジャッジ時間 6,944 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 101 ms
40,208 KB
testcase_01 AC 230 ms
47,052 KB
testcase_02 AC 223 ms
44,560 KB
testcase_03 AC 219 ms
44,600 KB
testcase_04 AC 110 ms
40,984 KB
testcase_05 AC 105 ms
40,664 KB
testcase_06 AC 110 ms
41,368 KB
testcase_07 AC 111 ms
41,384 KB
testcase_08 AC 148 ms
42,028 KB
testcase_09 AC 123 ms
41,460 KB
testcase_10 AC 128 ms
41,264 KB
testcase_11 AC 232 ms
46,608 KB
testcase_12 AC 223 ms
46,716 KB
testcase_13 AC 230 ms
46,676 KB
testcase_14 AC 220 ms
46,480 KB
testcase_15 AC 219 ms
46,096 KB
testcase_16 AC 220 ms
45,880 KB
testcase_17 AC 219 ms
46,300 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

package yukicoder;

import java.util.*;

public class Q430 {
	public static void main(String[] args) {
		new Q430().run();
	}

	void run() {
		solver();
	}

	void solver() {
		TrieByList trie=new TrieByList();
		Scanner sc=new Scanner(System.in);
		char[] s=sc.next().toCharArray();
		int m=sc.nextInt();
		while(m-->0){
			trie.add(sc.next().toCharArray());
		}
		trie.buildFailure();
		System.out.println(trie.countHit(s));
	}

	class TrieByList {
		Node root = new Node(0, (char) -1);
		int gen = 1;

		class Node {
			int id;
			char c;
			Node[] child=null;
			int ptn=0;
			int p=0;
			Node fail;
			int hit = 0;

			public Node(int id, char c) {
				this.id = id;
				this.c = c;
			}

			Node search(char c) {
				if (ptn << 31 - (c - 'a') < 0)
					return child[Integer.bitCount(ptn << 31 - (c - 'a')) - 1];
				else
					return null;
			}

			void append(Node n) {
				if (p == 0)
					child = new Node[1];
				else if (p + 1 > child.length) {
					child = Arrays.copyOf(child, child.length * 2);
				}

				int zind = Integer.bitCount(ptn << 31 - (n.c - 'a'));
				System.arraycopy(child, zind, child, zind + 1, p - zind);
				child[zind] = n;
				ptn |= 1 << (n.c - 'a');
				++p;
			}
		}

		void add(char[] s) {
			Node pre = null;
			Node cur = root;
			for (char c : s) {
				pre = cur;
				cur = pre.search(c);
				if (cur == null) {
					cur = new Node(gen++, c);
					pre.append(cur);
				}
			}
			++cur.hit;
		}

		void buildFailure() {
			root.fail = null;
			Queue<Node> q = new ArrayDeque<>();
			q.add(root);
			while (!q.isEmpty()) {
				Node cur = q.poll();
				outer: for (int i = 0; i < cur.p; ++i) {
					Node ch = cur.child[i];
					q.add(ch);
					for (Node to = cur.fail; to != null; to = to.fail) {
						Node next = to.search(ch.c);
						if (next != null) {
							ch.fail = next;
							ch.hit += next.hit;
							continue outer;
						}
					}
					ch.fail = root;
				}
			}
		}

		int countHit(char[] s) {
			int hit = 0;
			Node cur = root;
			outer:for (char c : s) {
				 for (Node to = cur; to != null; to = to.fail) {
					Node next = to.search(c);
					if (next != null) {
						hit += next.hit;
						cur = next;
						continue outer;
					}
				}
				cur = root;
			}
			return hit;
		}
	}
}
0