（每日一练c++）CC122 实现strstr函数

描述

实现函数 strStr。

函数声明如下：

char *strStr(char *str, char *dest)

返回一个指针，指向dest第一次在str中出现的位置，如果dest不是str的子串，则返回null

public class Solution {
    public String strStr(String haystack, String needle) {
        if(haystack==null || needle == null || needle.length()==0)
            return haystack;
        if(needle.length()>haystack.length())
            return null;
        char[] a = haystack.toCharArray();
		char[] b = needle.toCharArray();
        
		int[] next = getNextArray(b);
		int i = 0,j = 0;

        while (i < a.length) {
            while (j >= 0 && a[i] != b[j]) {
                j = next[j];
            }
            i++;
            j++;
            if (j == b.length) {
                return haystack.substring(i - b.length);
            }
        }
		return null;
    }
    private int[] getNextArray(char[] s){
		if(s.length==1)
			return new int[]{-1};
		int[] next = new int[s.length];
		next[0] = -1;
		next[1] = 0;
		int pos = 2;
		int cn = 0;
		while(pos0)
				cn = next[cn];
			else
				next[pos++] = 0;
		}
		return next;
	}
}