cdoj1091-秋实大哥の恋爱物语 【kmp】

http://acm.uestc.edu.cn/#/problem/show/1091

秋实大哥の恋爱物语

Time Limit: 5000/2000MS (Java/Others)     Memory Limit: 32000/32000KB (Java/Others)

Submit Status

传说有这么一个故事！

在一个月白风清的晚上，秋实大哥约一位他心仪的妹子一起逛校园，浪漫的秋实大哥决定在当晚对妹子表白。“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]，求解下一次出现情况，最后达到目的。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;

const int N=2000005;
int s[N],t[N];
int next_[N],len,ls;

void get_next_();
int kmp();

int main(){
    //freopen("D:\input.in","r",stdin);
    scanf("%d%d",&ls,&s[0]);
    for(int i=1;i<ls;i++)   scanf("%d",&s[i]),s[i-1]=s[i]-s[i-1];
    scanf("%d%d",&len,&t[0]);
    for(int i=1;i<len;i++)   scanf("%d",&t[i]),t[i-1]=t[i]-t[i-1];
    ls--,len--;
    get_next_();
    int ans=kmp();
    if(ans) printf("Wow! Life Winner!
%d
",ans);
    else    puts("Oh. That's impossible. I must have had a dream.");
    return 0;
}
void get_next_(){
    next_[0]=-1;
    next_[1]=0;
    for(int i=1;i<len;i++){
        int j=next_[i];
        while(1){
            if(t[i]==t[j]){
                next_[i+1]=j+1;
                break;
            }else{
                j=next_[j];
            }
            if(j<0){
                next_[i+1]=0;
                break;
            }
        }
    }
}
int kmp(){
    int ans=0,i=0,j=0;
    while(i<ls){
        if(j<0||s[i]==t[j])  ++i,++j;
        else    j=next_[j];
        if(j==len){
            ans++;
            j=next_[j];
        }
    }
    return ans;
}