洛谷P1628合并序列【模板】(Trie+dfs)

很久之前写的题了,当时不知道怎么dfs所以卡了一段时间，^_^

题解:由于题目给了一大堆字符串,所以首先考虑应该可以建树,之后找到T所在的位置,对T所在的位置dfs就行了

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;

int n,idx=0,x=0;
int son[N][26],cnt[N];
char str[110],tmp[110];
char ans[110];

void insert(char str[]){           //建树
    int p=0;
    for(int i=0;str[i];++i){
        int u=str[i]-'a';
        if(!son[p][u])  son[p][u]=++idx;
        p=son[p][u];
    }
    cnt[p]++;
}

void dfs(int u){                     //dfs查找
    while(cnt[u]--){
        cout<<tmp;
        for(int i=0;i<x;i++) cout<<ans[i];
        cout<<endl;
    }
    for(int i=0;i<26;i++){
        if(son[u][i]){
            int r=u;
            u=son[u][i];
            ans[x++]=i+'a';
            dfs(u);
            u=r;
            x--;
        }
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    while(n--){
        cin>>str;
        insert(str);
    }
    cin>>tmp;
    int p=0;
    for(int i=0;tmp[i];i++){
        int u=tmp[i]-'a';
        p=son[p][u];
    }
    dfs(p);
    return 0;
}