(POJ) 1816 Wild Words (字典树+搜索)

传送门

题目大意：给你n个模板字符串和m个待匹配字符串，模板字符串中有‘?’可以匹配任意一个字符，‘*’可以匹配0个1个或者多个字符串。问待匹配字符串可以和哪些模板字符串相匹配。

解题思路：我们可以将‘*’和‘?’分别放在26和27两个位置，存储字符串的时候，将每个字符串的最后一个节点返回存储起来，find不在和原来一样，而是一个搜索，当搜索的位置==len，而且当且节点为字符串的尾节点(flag标记)，那把vis置成true。然后对于‘*’，‘?’，和字符分别搜素下去。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int maxn=2e6+5;
int tree[maxn][30],tot=0;
bool flag[maxn],vis[maxn];
int n,m,p[maxn];
int add(char *s) {
	int root=0,id,len=strlen(s);
	for(int i=0; i<len; ++i) {
		if(s[i]=='*')	id=26;
		else if(s[i]=='?')	id=27;
		else id=s[i]-'a';
		if(!tree[root][id])	tree[root][id]=++tot;
		root=tree[root][id];
	}
	flag[root]=1;
	return root;
}
void find(char *s,int nroot,int pos) {
	int root=nroot,id,len=strlen(s);
	if(pos>len)	return;
	if(pos==len && flag[root]) {
		vis[root]=1;
	}
	if(tree[root][26]) { //有*
		for(int j=pos; j<=len; ++j) {
			find(s,tree[root][26],j);
		}
	}
	if(tree[root][27]) { //有？
		find(s,tree[root][27],pos+1);
	}
	id=s[pos]-'a';
	if(s[pos]>='a'&&s[pos]<='z'&&tree[root][id]) {
		find(s,tree[root][id],pos+1);
	}
}
char tp[50];
int main() {
	scanf("%d%d",&n,&m);
	for(int i=0; i<n; ++i) {
		scanf("%s",tp);
		p[i]=add(tp);
	}
	while(m--) {
		scanf("%s",tp);
		find(tp,0,0);
		bool flag=0;
		for(int i=0; i<n; ++i) {
			if(vis[p[i]]) {
				cout<<i<<" ";
				flag=1;
			}
		}
		if(!flag)	cout<<"Not match"<<endl;
		else	cout<<endl;
		for(int i=0; i<n; ++i)	vis[p[i]]=0;
	}
	return 0;
}