【bzoj4212】神牛的養成計劃 Trie樹+可持久化Trie樹
題目描述
Hzwer成功培育出神牛細胞,可最終培育出的生物體卻讓他大失所望...... 後來,他從某同校女神 牛處知道,原來他培育的細胞發生了基因突變,原先決定神牛特征的基因序列都被破壞了,神牛hzwer很生氣,但他知道基因突變的低頻性,說不定還有以下優秀基因沒有突變,那麽他就可以用限制性核酸內切酶把它們切出來,然後再構建基因表達載體什麽的,後面你懂的...... 黃學長現在知道了N個細胞的DNA序列,它們是若幹個由小寫字母組成的字符串。一個優秀的基因是兩個字符串s1和s2,當且僅當s1是某序列的前綴的同時,s2是這個序列的後綴時,hzwer認為這個序列擁有這個優秀基因。 現在黃學長知道了M個優秀基因s1和s2,它們想知道對於給定的優秀基因,有多少個細胞的DNA序列擁有它。輸入
輸出
輸出M行,每行一個數,第i行的數表示有多少個序列擁有第i個優秀基因。樣例輸入
10
emikuqihgokuhsywlmqemihhpgijkxdukjfmlqlwrpzgwrwozkmlixyxniutssasrriafu
emikuqihgokuorocifwwymkcyqevdtglszfzgycbgnpomvlzppwrigowekufjwiiaxniutssasrriafu
emikuqihgokuorociysgfkzpgnotajcfjctjqgjeeiheqrepbpakmlixyxniutssasrriafu
emikuqihgokuorociysgfrhulymdxsqirjrfbngwszuyibuixyxniutssasrriafu
emikuqihgokuorguowwiozcgjetmyokqdrqxzigohiutssasrriafu
emikuqihgokuorociysgvzfcdxdiwdztolopdnboxfvqzfzxtpecxcbrklvtyxniutssasrriafu
emikuqihgokuorocsbtlyuosppxuzkjafbhsayenxsdmkmlixyxniutssasrriafu
emikuqihgokuorociysgfjvaikktsixmhaasbvnsvmkntgmoygfxypktjxjdkliixyxniutssasrriafu
10
emikuqihgokuorociysg yxniutssasrriafu
aiegqmedckgqknky eqpoowonnewbq
xfbdnjbazhdnhkhvb qrqgbnmlltlkkbtyn
bjfhrnfedlhrlolzfv qppxpoofxcr
zhdfpldcbjf stsidponnvnmmdvap
zhdfpldcbjfpjmjxdt gdstsidponnvnmmdvap
dlhjtphgfnjtnqnbhxr wxwmhtsrrzrqqhzet
bjfhrnfedlhrlolzfv frqppxpoofxcr
zhdfpldcbjf dponnvnmmdvap
ucyakgyxweakehes nondykjiiqihhyqvk
樣例輸出
4
7
3
5
5
1
3
5
10
4
題解
Trie樹+可持久化Trie樹
一開始都蛋疼的想到樹套樹了 = =
先對所有字符串建立一棵Trie樹。
然後DFS一遍得到的Trie樹,按照DFS序將字符串的反串插入到可持久化Trie樹中,並記錄以每個節點為根的子樹中的字符串的範圍。
對於每個詢問,將在Trie樹中找s1,到達目標節點後可以直接拿出它包含字符串的範圍,再將s2反過來,根據這個範圍在可持久化Trie樹中查找有多少個反串以它的反串為前綴(即以它為後綴),得到答案。
註意判無解的情況。
#include <cstdio> #include <cstring> #include <algorithm> #define N 2000010 using namespace std; int next[26][N] , fa[N] , tag[N] , lp[N] , rp[N] , tot = 1 , cnt , root[N] , c[26][N] , si[N] , num , len; char str[N] , ch[N] , ss[N] , ll; void insert(int x , int &y) { int i , j , t = ++num; y = t; for(i = 0 ; i < len ; i ++ ) { for(j = 0 ; j < 26 ; j ++ ) c[j][t] = c[j][x]; c[str[i] - ‘a‘][t] = ++num , x = c[str[i] - ‘a‘][x] , t = c[str[i] - ‘a‘][t] , si[t] = si[x] + 1; } for(i = 0 ; i < 26 ; i ++ ) c[i][t] = c[i][x]; } int query(int x , int y) { int i; for(i = len - 1 ; ~i ; i -- ) x = c[str[i] - ‘a‘][x] , y = c[str[i] - ‘a‘][y]; return si[y] - si[x]; } void dfs(int x) { int i; if(tag[x]) { len = 0 , lp[x] = rp[x] = ++cnt; for(i = x ; i != 1 ; i = fa[i]) str[len ++ ] = ch[i]; insert(root[cnt - 1] , root[cnt]); return; } lp[x] = 1 << 30 , rp[x] = -1 << 30; for(i = 0 ; i < 26 ; i ++ ) if(next[i][x]) dfs(next[i][x]) , lp[x] = min(lp[x] , lp[next[i][x]]) , rp[x] = max(rp[x] , rp[next[i][x]]); } void trans(char *s , int ans) { int i , l = strlen(s); for(i = 0 ; i < l ; i ++ ) s[i] = (s[i] - ‘a‘ + ans) % 26 + ‘a‘; } int main() { int n , m , i , j , l , t , ans = 0; scanf("%d" , &n); for(i = 1 ; i <= n ; i ++ ) { scanf("%s" , str) , l = strlen(str); for(j = 0 , t = 1 ; j < l ; j ++ ) { if(!next[str[j] - ‘a‘][t]) next[str[j] - ‘a‘][t] = ++tot , ch[tot] = str[j] , fa[tot] = t; t = next[str[j] - ‘a‘][t]; } tag[t] = 1; } dfs(1); scanf("%d" , &m); for(i = 1 ; i <= m ; i ++ ) { scanf("%s%s" , ss , str) , trans(ss , ans) , trans(str , ans) , l = strlen(ss) , len = strlen(str); for(j = 0 , t = 1 ; t && j < l ; j ++ ) t = next[ss[j] - ‘a‘][t]; if(t) printf("%d\n" , ans = query(root[lp[t] - 1] , root[rp[t]])); else printf("%d\n" , ans = 0); } return 0; }
【bzoj4212】神牛的養成計劃 Trie樹+可持久化Trie樹