hdu 1867 KMP

来源:互联网 发布:linux宕机日志在哪里 编辑:程序博客网 时间:2024/05/29 18:18

http://acm.hdu.edu.cn/showproblem.php?pid=1867

题目大意,给定两个字符串,若一个字串的前缀是另一个字串的后缀,则可以合并。要求合并后的字串首先要尽可能短,然后要按字典序尽可能的短;

思路:两个字串按先后顺序不同可以有两种组合,对与每种组合得到KMP算法的next数组,next[len]就是两个字串可以合并的长度;

#include<iostream>#include<stdio.h>#include<string.h>#include<stdlib.h>#define min(a,b) a<b?a:busing namespace std;char s[200005],a[100005],b[100005];int next[200005];int KMP(int len,int n){int i=0,j=-1;next[0]=-1;while(i<n){if(j==-1||s[i]==s[j]){i++;j++;next[i]=j;}elsej=next[j];}while(next[n]>len){n=next[n];}return next[n];}int main(){while(scanf("%s%s",a,b)!=EOF){int len1=strlen(a);int len2=strlen(b);strcat(strcpy(s,a),b);int ans1=KMP(min(len1,len2),len1+len2);//第一个参数为可以合并的最大长度strcat(strcpy(s,b),a);int ans2=KMP(min(len1,len2),len1+len2);int i;if(ans1>ans2){for(i=0;i<len2;i++)printf("%c",b[i]);for(i=ans1;i<len1;i++)printf("%c",a[i]);}else if(ans2>ans1){for(i=0;i<len1;i++)printf("%c",a[i]);for(i=ans2;i<len2;i++)printf("%c",b[i]);}else{if(strcmp(a,b)>0){for(i=0;i<len2;i++)printf("%c",b[i]);for(i=ans1;i<len1;i++)printf("%c",a[i]);}else{for(i=0;i<len1;i++)printf("%c",a[i]);for(i=ans2;i<len2;i++)printf("%c",b[i]);}}printf("\n");}return 0;}


原创粉丝点击