Number Sequence
Time Limit: 10000/5000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 19913 Accepted Submission(s): 8535
題目大意:求b在a中第一次出現開始位置的下標(從1開始)
思路:kmp模板題,不過要注意b中可以出現負數,所有還是要改一點點的,因爲kmp模板解決的是字符串問題,這裏把字符串改成數組就OK了
<span style="font-size:18px;">#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#define LL long long
using namespace std;
const int maxn = 10000+5;
int Next[maxn],a[maxn*100],b[maxn],na,nb;
void getNext()
{
Next[0]=-1;
Next[1]=0;
for(int i=2;i<nb;i++){
if(b[i-1]==b[Next[i-1]]) Next[i]=Next[i-1]+1;
else{
int t=Next[i-1];
while(b[i-1]!=b[t]){
t=Next[t];
if(t==-1) break;
}
Next[i]=t+1;
}
}
}
int kmp()
{
getNext();
int i=0,j=0;
while(i<na && j<nb){
if(j==-1 || a[i]==b[j]) i++,j++;
else j=Next[j];
}
if(j==nb) return i-nb+1;
return -1;
}
int main()
{
int t,x,ans;
cin>>t;
while(t--){
cin>>na>>nb;
for(int i=0;i<na;i++) cin>>x,a[i]=x;
for(int i=0;i<nb;i++) cin>>x,b[i]=x;
ans=kmp();
cout<<ans<<endl;
}
return 0;
}
</span>