1. 程式人生 > >[Codeforces 961F] k-substrings

[Codeforces 961F] k-substrings

原題連結

Description

給定一個長度為n的字串S
我們設S的k-子串是S[k…n-k+1],設字串t是字串T的“奇正確前後綴“當且僅當滿足以下條件:

  1. t長度為奇數
  2. |t|<|T|
  3. t是T的border(既是字首又是字尾)
    對於k=1…n/2上取整,求S的k-子串的最長“奇正確前後綴“長度。無解輸出-1
    2≤n≤1,000,000

Solution

首先我們可以發現兩點性質

  • 設k=i的答案為F[i],那麼 F
    [ i + 1 ] F [ i ]
    2 F[i+1]\geq F[i]-2
    (去頭去尾即可)
  • 如果對於k=i,F[i]是可以的,F[i]-1不一定可以

那麼我們不妨考慮將對稱的極長串(即不能再向兩邊擴充套件)的那些找出來,然後更新它們左端點的那一個,再掃一邊,用F[i-1]-2更新F[i]即可

考慮怎麼找對稱的極長,因為要求了t的長度為奇數,那麼其必有一中心
我們列舉左邊的那個中心,對稱的那一個可以算出來

那此時向兩邊擴充套件就滿足二分性了(即i可以i-1一定可以),那麼直接二分+hash判就好

複雜度O(N log N)

Code

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define mo 1000000009
#define LL long long
#define N 1000005
using namespace std;
char ch[N];
int n,f[N];
LL h[N],K=119,cf[N];
bool pd(int p,int q,int le)
{
	int l=p-le+1,r=p+le-1,x=q-le+1,y=q+le-1;
	return ((h[r]-h[l-1]*cf[2*le-1]%mo+mo)%mo==(h[y]-h[x-1]*cf[2*le-1]%mo+mo)%mo);
}
int main()
{
	cin>>n;
	scanf("\n");
	cf[0]=1;
	fo(i,1,n) 
	{
		ch[i]=getchar();
		h[i]=(h[i-1]*K%mo+ch[i]-'a')%mo;
		cf[i]=cf[i-1]*K%mo;
	}
	int n1=(n+1)>>1;
	fo(i,1,n1)
	{
		if(n-i+1==i) break;
		int l=1,r=i;
		while(l+1<r)
		{
			int mid=(l+r)>>1;
			if(pd(i,n-i+1,mid)) l=mid;
			else r=mid;
		}
		if(pd(i,n-i+1,r)) f[i-r+1]=max(f[i-r+1],2*r-1);
		else if(pd(i,n-i+1,l)) f[i-l+1]=max(f[i-l+1],2*l-1);
	}
	fo(i,1,n1)
	{
		f[i]=max(f[i],f[i-1]-2);
		if(f[i]==0) printf("-1 ");
		else printf("%d ",f[i]);
	}
}