题意:
给两数组,求一个是否是另一个的子数组,若是返回匹配的首位置
分析:
KMP 入门
//扫描字符串A,并更新可以匹配到B的什么位置。 #include <map> #include <set> #include <list> #include <cmath> #include <queue> #include <stack> #include <cstdio> #include <vector> #include <string> #include <cctype> #include <complex> #include <cassert> #include <utility> #include <cstring> #include <cstdlib> #include <iostream> #include <algorithm> using namespace std; typedef pair<int,int> PII; typedef long long ll; #define lson l,m,rt<<1 #define pi acos(-1.0) #define rson m+1,r,rt<<11 #define All 1,N,1 #define read freopen("in.txt", "r", stdin) const ll INFll = 0x3f3f3f3f3f3f3f3fLL; const int INF= 0x7ffffff; int a[1000001],b[10001],next[10001]; void getNext(int m){ next[0]=-1; int i=0,j=-1; while(i<m){ if(j==-1||b[i]==b[j]){ i++; j++; if(b[i]==b[j])next[i]=next[j]; else next[i]=j; } else j=next[j]; } } void KMP(int n,int m){ int i=0,j=0; while(i<n&&j<m){ if(a[i]==b[j]||j==-1){ i++; j++; } else j=next[j]; } if(j==m){ printf("%d\n",i-m+1); } else printf("-1\n"); } int main() { int n,m,t; scanf("%d",&t); while(t--){ scanf("%d%d",&n,&m); for(int i=0;i<n;++i) scanf("%d",&a[i]); for(int j=0;j<m;++j) scanf("%d",&b[j]); if(m>n)printf("-1\n"); else { getNext(m); KMP(n,m); } } return 0; }
时间: 2024-11-08 23:54:39