LeetCode - 28. Implement strStr()

28. Implement strStr() #

Problem's Link
#

 ----------------------------------------------------------------------------

Mean: 

给定两个字符串str1和str2，输出str2在str1中第一次出现的下标.

analyse:

KMP算法模板题.

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-02-19-16.45
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long(LL);
typedef unsigned long long(ULL);
const double eps(1e-8);


class Solution
{
public:
   int Next[100005];
   void getNext(string& s)
   {
       Next[0]=0;
       for(int i=1,k=0;i<s.length();++i)
       {
           while(s[i]!=s[k] && k)
               k=Next[k-1];
           if(s[i]==s[k])
               ++k;
           Next[i]=k;
       }
   }
   int strStr(string haystack, string needle)
   {
       if(needle.length()==0)
           return 0;
       getNext(needle);
       for(int i=0,k=0;i<haystack.length();++i)
       {
           while(haystack[i]!=needle[k] && k)
               k=Next[k-1];
           if(haystack[i]==needle[k])
               ++k;
           if(k==needle.length())
               return i-k+1;
       }
       return -1;
   }
};

int main()
{
   string str1,str2;
   while(cin>>str1>>str2)
   {
       Solution solution;
       cout<<solution.strStr(str1,str2)<<endl;
   }
   return 0;
}
/*

*/