28. 实现 strStr() 函数

Posted 2021-09-01 sherry

 

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。如果不存在，则返回  -1 。

说明：

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。

示例 1：

输入：haystack = "hello", needle = "ll"
输出：2

示例 2：

输入：haystack = "aaaaa", needle = "bba"
输出：-1

 

class Solution {
    public int strStr(String haystack, String needle) {

        if(haystack==null||needle==null
           ||needle.length()>haystack.length()){
            return -1;
        }
        if(needle==""||needle.length()<1){
            return 0;
        }
        
     int[] nexts=getNextArrary(needle.toCharArray());
     char[] s1=haystack.toCharArray();
     char[] s2=needle.toCharArray();
     int x=0;
     int y=0;   
     while(s1.length>x&&s2.length>y){
         if(s1[x]==s2[y]){
             x++;
             y++;
         }else if(nexts[y]==-1){
             x++;
         }else{
             y=nexts[y];
         }
     } 
     return y==s2.length?x-y:-1;   
        
        
    }
    
    public int[] getNextArrary(char[] chs){
        if(chs.length==1){
            return new int[]{-1};
        }
        int[] nexts=new int[chs.length];
        nexts[0]=-1;
        nexts[1]=0;
        int i=2;
        int y=0;
        while(i<chs.length){
            if(chs[i-1]==chs[y]){
                nexts[i++]=++y;
            }else if(y>0){
                y=nexts[y];
            }else{
                nexts[i++]=0;
            }
            
        }
        return nexts;
        
    }
}