kmjp's blog

競技プログラミング参加記です

yukicoder : No.962 LCPs

これは定番っぽいしどっかで解いてそう。
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コンお疲れ様でした。
まだ解けてない問題もあるのでおいおい解いていきます。