1. 程式人生 > >bzoj4567: [Scoi2016]背單詞

bzoj4567: [Scoi2016]背單詞

sed tchar freopen aps 技術 lld scanf zoj etc

d1t1,比較良心。雖然感覺我寫得比較鬼畜。

容易看出一種貪心的思路,把串按後綴關系分組,不同組分開考慮,同組的放一塊兒寫,然後組的size小的靠前寫是一個最優的方案。

同組的也可以同樣遞歸下去考慮。

於是先建出字典樹,然後把後綴關系的樹建出來,在這顆樹上貪心。

註意需要在這顆後綴關系樹上貪心(把兒子按size大小排序),不能直接在字典樹上貪心(一個size大的的兒子不一定是同一組後綴關系)。

技術分享圖片
//Achen
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include
<vector> #include<cstdio> #include<queue> #include<cmath> const int N=1010007; typedef long long LL; using namespace std; int n; char s[N]; LL ans; template<typename T>void read(T &x) { char ch=getchar(); x=0; T f=1; while(ch!=-&&(ch<0||ch>9)) ch=getchar();
if(ch==-) f=-1,ch=getchar(); for(;ch>=0&&ch<=9;ch=getchar()) x=x*10+ch-0; x*=f; } int ch[N][26],is[N],sz[N],rt,tot; void insert(int len) { int x=rt; for(int i=len-1;i>=0;i--) { int c=s[i]-a; if(!ch[x][c]) ch[x][c]=++tot; x=ch[x][c]; }
is[x]=1; } bool cmp(const int &A,const int &B) { return sz[A]<sz[B]; } vector<int>son[N]; void add(int u,int v) { son[u].push_back(v); } int pos,p[N]; void dfs1(int x,int fa) { if(is[x]) add(fa,x); for(int i=0;i<26;i++) if(ch[x][i]) dfs1(ch[x][i],is[x]?x:fa); } void dfs2(int x) { int up=son[x].size(); sz[x]=1; for(int i=0;i<up;i++) { int y=son[x][i]; dfs2(y); sz[x]+=sz[y]; } } void dfs3(int x,int pr) { sort(son[x].begin(),son[x].end(),cmp); int up=son[x].size(); if(x!=0) ++pos; int tp=pos; ans+=tp-pr; for(int i=0;i<up;i++) { int y=son[x][i]; dfs3(y,tp); } } int main() { //freopen("1.in","r",stdin); read(n); for(int i=1;i<=n;i++) { scanf("%s",s); int len=strlen(s); insert(len); } dfs1(rt,0); dfs2(rt); dfs3(rt,0); printf("%lld\n",ans); return 0; }
View Code

bzoj4567: [Scoi2016]背單詞