cdoj1091-秋实大哥の恋爱物语 【kmp】
http://acm.uestc.edu.cn/#/problem/show/1091
秋实大哥の恋爱物语
Time Limit: 5000/2000MS (Java/Others) Memory Limit: 32000/32000KB (Java/Others)
传说有这么一个故事!
在一个月白风清的晚上,秋实大哥约一位他心仪的妹子一起逛校园,浪漫的秋实大哥决定在当晚对妹子表白。“XXXXX...”,秋实大哥温情地说完了准备已久的话。而妹子决定用一种浪漫的方式接受秋实大哥(其实妹子早已对秋实大哥动心,这一刻她早已迫不及待了,但还是决定考秋实大哥最后一关,再委婉地接受)。妹子拿出了她心爱的口琴,吹出了一首迷人的曲子...... “你能把我的曲子重复一遍么?”,但考虑到万一秋实大哥没有做到而失去了赢得人赢的心的机会,妹子又说到,“只要你能吹出我的一部分旋律,我就答应你,从今以后,我就是你的一部分”。
好奇心重的你,真的很想知道秋实大哥最终有没有抱得美人归,除此之外,你还想知道秋实大哥吹出的曲子的旋律有多少次符合妹子的旋律。
将两个相邻的音符连起来,则妹子吹出的音符可以画出一条折线A,同样,秋实大哥吹出的音符也可以画出一条折线B,如果折线B已经与折线A的某一段完全重合,或者能够经过上下左右平移与折线A的某一段完全重合,则表示秋实大哥吹出了妹子的一部分旋律。
Input
第一行输入一个整数N(2≤N≤2 000 000),表示妹子吹了N个音符。
第二行输入N个音符,每个音符都是整数,且在32位整数范围内,每两个音符用一个空格隔开。
第三行输入一个整数M(2≤M≤2 000 000),表示秋实大哥吹了M个音符。
最后一行输入M个音符,每个音符都是整数,且在32位整数范围内,每两个音符用一个空格隔开。
Output
如果秋实大哥抱得美人归了,第一行输出Wow! Life Winner!
,第二行再输出一个整数,表示秋实大哥的曲子的旋律有多少次符合妹子的。
如果秋实大哥没有做到,输出Oh. That's impossible. I must have had a dream.
Sample input and output
Sample Input | Sample Output |
---|---|
14 1 1 5 5 6 6 5 4 4 3 3 2 2 1 14 0 0 4 4 5 5 4 3 3 2 2 1 1 0 |
Wow! Life Winner! 1 |
20 1 2 1 2 1 2 1 2 1 1 0 1 3 2 3 2 7 6 7 2 3 6 5 6 |
Wow! Life Winner! 6 |
25 2 3 2 3 3 2 3 3 3 2 3 2 2 3 3 2 2 2 3 3 3 3 2 3 3 3 2 3 3 |
Wow! Life Winner! 5 |
29 6 6 7 5 5 6 4 4 5 3 3 4 2 2 3 1 1 2 0 0 1 -1 -1 0 -2 -2 -1 -3 -3 8 6 6 7 5 5 6 4 4 |
Wow! Life Winner! 8 |
26 1 1 0 1 0 1 0 1 0 1 1 0 1 0 1 0 1 1 0 1 0 1 1 0 1 0 5 1 1 0 1 1 |
Oh. That's impossible. I must have had a dream. |
解题思路:kmp。题目说判断折线重合,可以把原数据的相邻的点做减法,就可以直接用kmp求子串出现次数了。至于kmp,可以参考我这篇博文:http://www.cnblogs.com/jiu0821/p/4189987.html。单求子串第一次出现的位置,不需要用到next数组的next[len];求子串出现次数的话,需要用到next[len],意思是每次匹配完毕,j就等于len,这时候理解为第len项不匹配(事实上t串没有这一项),令j=next[len],求解下一次出现情况,最后达到目的。
代码:
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cstdlib> 5 using namespace std; 6 7 const int N=2000005; 8 int s[N],t[N]; 9 int next_[N],len,ls; 10 11 void get_next_(); 12 int kmp(); 13 14 int main(){ 15 //freopen("D:\\input.in","r",stdin); 16 scanf("%d%d",&ls,&s[0]); 17 for(int i=1;i<ls;i++) scanf("%d",&s[i]),s[i-1]=s[i]-s[i-1]; 18 scanf("%d%d",&len,&t[0]); 19 for(int i=1;i<len;i++) scanf("%d",&t[i]),t[i-1]=t[i]-t[i-1]; 20 ls--,len--; 21 get_next_(); 22 int ans=kmp(); 23 if(ans) printf("Wow! Life Winner!\n%d\n",ans); 24 else puts("Oh. That's impossible. I must have had a dream."); 25 return 0; 26 } 27 void get_next_(){ 28 next_[0]=-1; 29 next_[1]=0; 30 for(int i=1;i<len;i++){ 31 int j=next_[i]; 32 while(1){ 33 if(t[i]==t[j]){ 34 next_[i+1]=j+1; 35 break; 36 }else{ 37 j=next_[j]; 38 } 39 if(j<0){ 40 next_[i+1]=0; 41 break; 42 } 43 } 44 } 45 } 46 int kmp(){ 47 int ans=0,i=0,j=0; 48 while(i<ls){ 49 if(j<0||s[i]==t[j]) ++i,++j; 50 else j=next_[j]; 51 if(j==len){ 52 ans++; 53 j=next_[j]; 54 } 55 } 56 return ans; 57 }