思路:
好题,必须记一笔。
这题有两个之前没有接触过的字典树新变形。
第一个是insert的方式,不再只是单单的将一整个单词都插进去,而是从单词的第一个字母开始往后遍历,依次插入第i个字母到末尾的单词,这样做就可以满足题目中“substring”的要求,即将“substring”的问题转换为我们熟悉的字典树可以解决的“前缀”问题。
第二个是标记每个char* s所属的原始字符串,这种做法可以避免一个单词有相同部分的问题。
AC代码:
#include#include using namespace std;struct node{ int num; int id; struct node* next[26]; node() { num = 0; id = -1; for(int i = 0;i < 26;i++) next[i] = NULL; }};node* root;char str[10007][27];void Insert(char* s,int mark){ node* p = root; for(;*s!='\0';s++) { int n = *s-'a'; if(p->next[n] == NULL) { p->next[n] = new node(); p = p->next[n]; p->id = mark; p->num = 1; } else { p = p->next[n]; if(p->id != mark) { p->id = mark; p->num++; } } }}int search(char* s){ node* p = root; for(;*s!='\0';s++) { int n = *s-'a'; if(p->next[n] == NULL) return 0; p = p->next[n]; } return p->num;}int main(){ root = new node(); int P; cin>>P; for(int i = 0;i < P;i++) { cin>>str[i]; char* tmp = str[i]; for(;*tmp!='\0';tmp++) Insert(tmp,i); } int tc; cin>>tc; char t[27]; while(tc--) { cin>>t; cout< <