Leetcode 28 实现strStr
这道题,最开始我想图省事,用了动态规划写…结果73 / 74 个通过测试用例。我觉得这道题应该题意就不想让DP做。最后一个测试用例超出了内存的限制。但是我既然写出来了,就贴一下代码><
class Solution {
public int strStr(String haystack, String needle) {
if(needle.length()==0) return 0;
int index=0,max=0;
int [][]dp=new int[haystack.length()+1][needle.length()+1];
bgm:for(int i=1;i<haystack.length()+1;i++)
{
for(int j=1;j<needle.length()+1;j++)
{
if(haystack.charAt(i-1)==needle.charAt(j-1))
{
dp[i][j]=1+dp[i-1][j-1];
if(max<dp[i][j])
{
max=dp[i][j];
index=i;
if(max==needle.length()) break bgm;
}
}
}
}
if(max<needle.length()) return -1;
else return index-max;
}
}
然后我就去搜别人的解答,发现大家都没有提到用DP…感觉自己out了。
KMP算法是应该是这道题要考察的知识点。其实这个算法我之前见过,也学过,但是基本没用过,所以当时完全没想到用这个办法(或者说我根本没掌握好,不以为意
KMP算法(研究总结,字符串) 看了这篇文章,觉得讲得非常到位了。(下面是对这篇文章的引用)
这里面涉及到两个代码段,一个是next数组的创建,还有一个就是next数组与原数组的匹配。
next数组创建(假设匹配数组B的长度为m)。令初始值为-1是为了方便判越界,同时为了方便判断第0位与第i位
next[0]=-1;
for (int i=1;i<m;i++)
{
int j=next[i-1];
while ((B[j+1]!=B[i])&&(j>=0))
j=next[j];
if (B[j+1]==B[i])
next[i]=j+1;
else
next[i]=-1;
}
假设被匹配数组A长度为n
int i=0;
while (i<n)
{
if (A[i]==B[j])
{
i++;
j++;
if (j==m)
{
println(i-m);//注意,这里输出的位置是从1开始标号的,如果你要输出从0开始标号的位置,应该是是i-m.
j=next[j-1]+1;
}
}
else
{
if (j==0)
i++;
else
j=next[j-1]+1;
}
}
Leetcode28代码
class Solution {
public int strStr(String haystack, String needle) {
if(needle.length()==0) return 0;
//f数组
int []F=new int[needle.length()];
F[0]=-1;
for (int i=1;i<needle.length();i++)
{
int j=F[i-1];
while ((needle.charAt(j+1)!=needle.charAt(i))&&(j>=0)) j=F[j];
if (needle.charAt(j+1)==needle.charAt(i)) F[i]=j+1;
else F[i]=-1;
}
int i=0,j=0;
while (i<haystack.length())
{
if (haystack.charAt(i)==needle.charAt(j))
{
i++;
j++;
if (j==needle.length())
{
return i-needle.length();
}
}
else
{
if (j==0)
i++;
else
j=F[j-1]+1;
}
}
return -1;
}
}