字典树Trie

第13次提交终于0->100的原因居然是——没有判重？？？！

 

 

https://www.luogu.org/problem/P3879算是道略显恶心的板子题吧，唯一一点以后可以继承的经验是trie数组利用了结构体+vector的方式解决了空间与维度的问题。

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std; 

struct Node{
    int a[26];
    Node(){memset(a,0,sizeof a);}
    int& operator [](int x){return a[x];}
}zero;

string s;
vector<Node> trie;
vector<int> cntw[500010];

void add(int i){
    int p = 0;
    for(int j = 0;j < s.size();j++){
        int ne = s[j]-'a';
        if(!trie[p][ne]){
            trie[p][ne] = trie.size();
            trie.push_back(zero);
        }
        p = trie[p][ne];
    }
    if(cntw[p].empty()||cntw[p].size()&&i^cntw[p][cntw[p].size()-1])
        cntw[p].push_back(i);
}

void work(){
    int flag = 1,p = 0;
    for(int i = 0;i < s.size();++i){
        int ne = s[i]-'a';
        if(!trie[p][ne]){flag = 0;break;}
        p = trie[p][ne];
    }
    if(flag)for(int i = 0;i < cntw[p].size();++i)
        printf("%d ",cntw[p][i]);
    putchar('\n');
}

int n,m;

int main(){
    trie.push_back(zero);
    cin >> n;
    for(int i = 1;i <= n;i++){
        int x;cin >> x;
        for(int j = 1;j <= x;j++){
            cin >> s;
            add(i);
        }
    }
    cin >> m;
    for(int i = 1;i <= m;i++){
        cin >> s;
        work();
    }    
return 0;
}