Trie+拓撲排序——Luogu3065 [USACO12DEC]第一!First!

題面:Luogu3065
我們首先考慮一種情況,如果某一字符串的某個前綴是另外一個字符串,這個字符串不可能字典序最小。
所以我們來考慮相同前綴的問題。如果某一字符串字典序最小,和它同前綴的字符串的相同前綴之後一位字母的大小順序就可以確定。如果這一系列的關係沒有矛盾的話,這個串就可以是最小的,反之不行。
判斷有無矛盾的話我們可以通過建連邊跑拓撲排序解決。至於找前綴這種問題,交給Trie樹就好了。

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <ctime>
#include <map>
#include <queue>
#include <cstdlib>
#include <string>
#include <climits>
#include <set>
#include <vector>
using namespace std;
bool b[300010];
int n,nedge=0,p[2010],nex[2010],head[2010],rd[2010];
int nxt[300010][26],ans=0,cnt=0,anss[200010];
char s[30010][310];
inline void pre(){
    nedge=0;memset(nex,0,sizeof nex);
    memset(head,0,sizeof head);memset(rd,0,sizeof rd);
}
inline void addedge(int a,int b){
    p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;
}
inline void insert(int x){
    int len=strlen(s[x]+1),now=0;
    for(int i=1;i<=len;i++){
        int p=s[x][i]-'a';
        if(!nxt[now][p])nxt[now][p]=++cnt;
        now=nxt[now][p];
    }
    b[now]=1;
}
inline bool check(int x){
    int len=strlen(s[x]+1),now=0;
    for(int i=1;i<=len;i++){
        int p=s[x][i]-'a';
        if(b[now])return 0;
        for(int i=0;i<26;i++)if(p!=i&&nxt[now][i])addedge(p+1,i+1),rd[i+1]++;
        now=nxt[now][p];
    }
    return 1;
}
inline bool topo(){
    queue<int>q;for(int i=1;i<=26;i++)if(!rd[i])q.push(i);
    while(!q.empty()){
        int now=q.front();q.pop();
        for(int k=head[now];k;k=nex[k])if(rd[p[k]]){
            rd[p[k]]--;if(!rd[p[k]])q.push(p[k]);
        }
    }
    for(int i=1;i<=26;i++)if(rd[i])return 0;
    return 1;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s[i]+1);
        insert(i);
    }
    for(int i=1;i<=n;i++){
        pre();
        if(check(i)&&topo())anss[++ans]=i;
    }
    printf("%d\n",ans);
    for(int i=1;i<=ans;i++)printf("%s\n",s[anss[i]]+1);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章