[P3065 [USACO12DEC] First! G (字典樹+判環)

Fire_Raku發表於2024-06-29

P3065 [USACO12DEC] First! G

字典樹+判環

考慮字典序的比較過程,什麼時候需要確定字母的順序?當有多個字串字首相同,比較當前位時需要字母的順序。於是可以建出字典樹,對於每個字串,單獨判斷是否可行。遍歷當前字串在字典樹上的路徑,如果一個節點下有多條邊連出,那麼就有若干個限制條件。考慮建圖,跑拓撲排序或者 dfs 判環即可。

複雜度 \(\Theta(\sum s+26n)\)

#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back

using i64 = long long;
using ull = unsigned long long;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 3e5 + 10;
int n, tot, ans;
bool vis[30010], ins[N], mp[30][30];
int cnt[N];
int tr[N][27];
std::string s[30010];
std::vector<int> e[30];
void insert(std::string s) {
	int u = 0;
	for(int i = 0; i < s.length(); i++) {
		int c = s[i] - 'a';
		if(!tr[u][c]) tr[u][c] = ++tot;
		u = tr[u][c];
	}
	cnt[u]++;
}
void init(std::string s) {
	int u = 0;
	for(int i = 0; i < s.length(); i++) {
		int c = s[i] - 'a';
		for(int j = 0; j < 26; j++) {
			if(c != j && tr[u][j] && !mp[c][j]) e[c].pb(j), mp[c][j] = 1;
		}
		u = tr[u][c];
	}
}

bool dfs(int u) {
	ins[u] = 1;
	for(auto v : e[u]) {
		if(ins[v] || dfs(v)) return 1;
	}
	ins[u] = 0;
	return 0;
}

void del(std::string s) {
	memset(ins, 0, sizeof(ins));
	memset(mp, 0, sizeof(mp));
	for(int i = 0; i < s.length(); i++) {
		int c = s[i] - 'a';
		e[c].clear();
	}
} 

bool check(std::string s) {
	int u = 0;
	for(int i = 0; i < s.length(); i++) {	
		int c = s[i] - 'a';
		if(cnt[u]) return 1;
		u = tr[u][c];
	}
	return 0;
}
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
	std::cin >> n;

	for(int i = 1; i <= n; i++) {
		std::cin >> s[i];
		insert(s[i]);
	}

	for(int i = 1; i <= n; i++) {
		if(check(s[i])) continue;
		init(s[i]);
		bool flg = 1;
		for(int j = 0; j < 26; j++) {
			if(dfs(j)) {
				flg = 0;
				break;
			}
		}
		if(flg) vis[i] = 1, ans++;
		del(s[i]);
	}

	std::cout << ans << "\n";

	for(int i = 1; i <= n; i++) if(vis[i]) std::cout << s[i] << "\n";

	return 0;
}

相關文章