【BZOJ4567】[Scoi2016]背單詞 Trie樹+貪心
阿新 • • 發佈:2017-05-23
字母 如果 ems scanf 序號 data scan name rdquo
輸入一個整數 n ,表示 Lweb 要學習的單詞數。接下來 n 行,每行有一個單詞(由小寫字母構成,且保證任意單
詞兩兩互不相同)1≤n≤100000, 所有字符的長度總和 1≤|len|≤510000
a
ba
【BZOJ4567】[Scoi2016]背單詞
Description
Lweb 面對如山的英語單詞,陷入了深深的沈思,“我怎麽樣才能快點學完,然後去玩三國殺呢?”。這時候睿智 的鳳老師從遠處飄來,他送給了 Lweb 一本計劃冊和一大缸泡椒,他的計劃冊是長這樣的: ————— 序號 單詞 ————— 1 2 …… n-2 n-1 n ————— 然後鳳老師告訴 Lweb ,我知道你要學習的單詞總共有 n 個,現在我們從上往下完成計劃表,對於一個序號為 x 的單詞(序號 1...x-1 都已經被填入): 1) 如果存在一個單詞是它的後綴,並且當前沒有被填入表內,那他需要吃 n×n 顆泡椒才能學會; 2) 當它的所有後綴都被填入表內的情況下,如果在 1...x-1 的位置上的單詞都不是它的後綴,那麽你吃 x 顆泡 椒就能記住它; 3) 當它的所有後綴都被填入表內的情況下,如果 1...x-1的位置上存在是它後綴的單詞,所有是它後綴的單詞中 ,序號最大為 y ,那麽你只要吃 x-y 顆泡椒就能把它記住。 Lweb 是一個吃到辣辣的東西會暴走的奇怪小朋友,所以請你幫助 Lweb ,尋找一種最優的填寫單詞方案,使得他 記住這 n 個單詞的情況下,吃最少的泡椒。Input
Output
Lweb 吃的最少泡椒數
Sample Input
2a
ba
Sample Output
2題解:我們將單詞倒序扔到Trie樹中去,然後根據貪心可知如果想填入一個單詞,那麽它父親節點上的單詞一定已經被填入,現在的問題就是我們如何處理一個父親的多個兒子的填入順序。
又根據貪心可知,如果一個單詞有多個兒子,那麽一定先走子樹最小的那個兒子,這是可證明的,但我懶啊~
#include <cstdio> #include <cstring> #include <iostream> #include <queue> #include <algorithm> using namespace std; int n,tot,cnt,now; long long ans; const int maxn=100010; struct node { int ch[26],v,sm,from; }p[500010]; char str[500010]; int fa[maxn],next[maxn],head[maxn],to[maxn],pos[maxn],q[maxn],siz[maxn],s[maxn],Q[maxn]; void add(int a,int b) { to[cnt]=b,next[cnt]=head[a],head[a]=cnt++; } void dfs1(int x) { siz[x]=1; for(int i=head[x];i!=-1;i=next[i]) dfs1(to[i]),siz[x]+=siz[to[i]]; } bool cmp(int a,int b) { return siz[a]<siz[b]; } void dfs2(int x) { int i,sum=0; ans+=s[x]-s[fa[x]]; for(i=head[x];i!=-1;i=next[i]) q[++sum]=to[i]; sort(q+1,q+sum+1,cmp); for(sum=0,i=head[x];i!=-1;i=next[i]) to[i]=q[++sum]; for(i=head[x];i!=-1;i=next[i]) s[to[i]]=++now,dfs2(to[i]); } int main() { scanf("%d",&n); int i,j,a,b,u; tot=1; for(i=1;i<=n;i++) { scanf("%s",str); a=strlen(str); for(u=1,j=a-1;j>=0;j--) { b=str[j]-‘a‘; if(!p[u].ch[b]) p[u].ch[b]=++tot,p[p[u].ch[b]].from=u; u=p[u].ch[b]; } p[u].v=i,pos[i]=u;; } memset(head,-1,sizeof(head)); for(i=1;i<=n;i++) { u=p[pos[i]].from; while(u!=1&&!p[u].v) u=p[u].from; fa[i]=p[u].v,add(fa[i],i); } dfs1(0),dfs2(0); printf("%lld",ans); return 0; }
【BZOJ4567】[Scoi2016]背單詞 Trie樹+貪心