[HDOJ 3522][Vjudge 10889] Minimum Integer sequence [扩展kmp]

给定两个字符串，将第二个字符串插入到第一个字符串的某个位置，求字典序最小的方案

进行一次扩展kmp，求出第二个字符串的某个位置开始和第二个字符串的最长公共前缀，求出第一个字符串的某个位置开始和第二个字符串的最长公共前缀

枚举每一种插入方法，然后用扩展kmp的结果找到第一个不同字符，比较得出谁的字典序更小

每次比较的复杂度是O(1)

#include <cstdio>
#include <cstring>

void exkmp(char s1[],char s2[],int next[],int ex[]) {
	int i,j,p;
	for (i=0,j=0,p=-1;s1[i]!='\0';i++,j++,p--) {
		if (p==-1) {
			j=0;
			do p++; while (s1[i+p]!='\0'&&s1[i+p]==s2[j+p]);
			ex[i]=p;
		} else if (next[j]<p) ex[i]=next[j];
		else if (next[j]>p) ex[i]=p;
		else {
			j=0;
			while (s1[i+p]!='\0'&&s1[i+p]==s2[j+p]) p++;
			ex[i]=p;
		}
	}
	ex[i]=0;
}

char s1[100100];
char s2[100100];
int ex[100100];
int next[100100];
int ls1,ls2;

bool smallerThan(int i,int j) {
	if (ex[j]<i-j) return s1[j+ex[j]]<s2[ex[j]];
	if (next[i-j]<ls2-(i-j)) return s2[next[i-j]]<s2[i-j+next[i-j]];
	if (next[ls2-(i-j)]<(i-j)) return s2[ls2-(i-j)+next[ls2-(i-j)]]<s2[next[ls2-(i-j)]];
	return false;
}

int main() {
	int ans,i;
	while (scanf("%s%s",s1,s2)!=EOF) {
		exkmp(s2+1,s2,next,next+1);
		exkmp(s1,s2,next,ex);
		ls1=strlen(s1);
		ls2=strlen(s2);
		ans=0;
		for (i=0;i<ls1;i++) {
			if (smallerThan(i+1,ans)) ans=i+1;
		}
		for (i=0;i<ans;i++) printf("%c",s1[i]);
		printf("%s",s2);
		for (i=ans;i<ls1;i++) printf("%c",s1[i]);
		printf("\n");
	}
	return 0;
}