bzoj4567: [Scoi2016]背單詞
阿新 • • 發佈:2018-02-03
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]背單詞