求大佬帮忙解释,为什么全wa
查看原帖
求大佬帮忙解释,为什么全wa
1300671
Halen1happy楼主2024/12/15 08:14
#include<bits/stdc++.h>
using namespace std;
//字典树的基本操作:查找,插入,删除
//字典树的应用 串的快速检索,串排序,前缀统计及判断,最长公共前序
//trie树是字典树 
const int N=3000005;
int son[N][65],cnt[N],idx;
char str[N];
int ge(char x){
    if(x>='A'&&x<='Z')
        return x-'A';
    else if(x>='a'&&x<='z')
        return x-'a';
    else
        return x-'0';
}
//插入字符串 
void insert(char *str){
	int p=0;
	int len=strlen(str);
	for(int i=0;i<len;i++){
		int u=ge(str[i]);
		if(!son[p][u])
		 son[p][u]=++idx;
		p=son[p][u];
	}
	cnt[p]++; 
}
//查询字符串
int query(char *str){
	int p=0,sum=0;
	int len=strlen(str);
	for(int i=0;i<len;i++){
		int u=ge(str[i]);
		if(cnt[p]){
			sum++;
		}
		if(!son[p][u]) 
		return 0;
		p=son[p][u];
	}
	if(cnt[p]) sum++;
	return sum;
} 
int main(){
		idx=0;
		int n,m;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++){
	    	scanf("%s",str);
			insert(str);
    	}
		for(int i=1;i<=m;i++){
			scanf("%s",str);
			printf("%d\n",query(str));
		}	
	
	return 0;
}
2024/12/15 08:14
加载中...