これは定番っぽいしどっかで解いてそう。
https://yukicoder.me/problems/no/962
問題
N個の文字列S[i]が与えられる。
区間[L,R]においてS[L]~S[R]のLCPがaの時、a*(R-L+1)だけスコアが入るとする。
この文字列群に対し総スコアを求めよ。
解法
ある区間[L...R]のprefixが何文字か一致しており、もう1文字追加で一致するなら(R-L)*(R-L+1)/2だけスコアが伸びる。
dfs(c,L,R) : c文字目まで[L,R]が一致しているとき、これらの(c+1)文字目の一致状況を見る。最大で[L',R']で一致するなら、(R'-L')*(R'-L'+1)/2だけスコアを追加し、dfs(c+1,L',R')を処理する。
というようにDFSしていくと、O(N+sum(S))程度で処理できる。
int N; string S[202020]; ll ret=0; ll sum[202020]; void dfs(int cur,int L,int R) { if(L>=R) return; int pre=L; int i; for(i=L;i<R;i++) { if(S[i].size()<=cur) { dfs(cur+1,pre,i); pre=i+1; } else { if(S[i][cur]!=S[pre][cur]) { dfs(cur+1,pre,i); pre=i; } ret+=sum[i-pre+1]; } } dfs(cur+1,pre,R); } void solve() { int i,j,k,l,r,x,y; string s; cin>>N; FOR(i,N) cin>>S[i]; for(i=1;i<=200000;i++) sum[i]=sum[i-1]+i; dfs(0,0,N); cout<<ret<<endl; }
まとめ
AdventCalendarコンお疲れ様でした。
まだ解けてない問題もあるのでおいおい解いていきます。