hihocoder 1260 String Problem I (Trie樹 好題)

3 3
tourist
petr
rng
toosimple
rg
ptr

0
1
1

題目分析：感覺正解不是Trie但是還是用Trie搞過了，其實就是在trie樹上DFS的時候有一個點可以“失配”而直接跳到下一層，Search(char *s, int pos, int p, bool ok)，s表示要查詢的字串，pos表示列舉到s的第pos位，p表示當前父親編號，ok表示之前是否已經跳過一次，true表示沒跳過，這裡有個問題就是比如

2 1
aa
ab
a

錯誤答案跑出來會是3，因為對於aa來說，他先跳和後跳的情況都被計算了，因此還需要記錄一個單詞是否已經滿足情況，由於點的個數太多但是n比較小，所以考慮將單詞結尾標號用map離散化，還有一點要注意的就是當pos==len的時候不能直接return，因為有可能在最後一個單詞那裡跳，比如

1 1

ab

b

所以當pos==len時要先判斷跳不跳，再return

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
int const MAX = 1e5 + 5;
int n, m, ans, len;
char s[MAX];

struct Trie
{
	int next[MAX * 26][26], end[MAX * 26], tot, root, num;
	bool vis[10005];
	map <int, int> mp;

	inline int Newnode()
	{
		memset(next[tot], -1, sizeof(next[tot]));
		end[tot] = false;
		return tot ++;
	} 

	inline void Init()
	{
		num = 0;
		tot = 0;
		root = Newnode();
	}

	inline void Insert(char *s)
	{
		int p = root;
		for(int i = 0; i < (int)strlen(s); i++)
		{
			int idx = s[i] - 'a';
			if(next[p][idx] == -1)
				next[p][idx] = Newnode();
			p = next[p][idx];
		}
		end[p] = true;
		mp[p] = num ++;
	}

	inline void Search(char *s, int pos, int p, bool ok)
	{
		if(end[p] && !vis[mp[p]] && pos == len)
		{
			vis[mp[p]] = true;
			ans ++;
			return;
		}
		if(ok)
			for(int i = 0; i < 26; i++)
				if(next[p][i] != -1)
					Search(s, pos, next[p][i], false);
		if(pos == len)
			return;
		int idx = s[pos] - 'a';
		if(next[p][idx] != -1)
			Search(s, pos + 1, next[p][idx], ok);

	}

}t;

int main()
{
	t.Init();
	scanf("%d %d",&n, &m);
	for(int i = 0; i < n; i++)
	{
		scanf("%s", s);
		t.Insert(s);
	}
	for(int i = 0; i < m; i++)
	{
		scanf("%s", s);
		ans = 0;
		len = strlen(s);
		memset(t.vis, false, sizeof(t.vis));
		t.Search(s, 0, t.root, true);
		printf("%d\n", ans);
	}
}