魔鏡啊魔鏡…

題目描述

給定一個含n個數的序列A和一個含m (m<=n) 個數的序列B。

詢問在A中有多少段連續的長為m的子序列Ak,Ak+1,…,Ak+m-1使得對于任意1<=i, j<=m滿足Ak+i-1-Bi=Ak+j-1-Bj

輸入

第一行兩個整數n，m (1 <= m <= n <= 106)

接下來一行n個整數，描述序列A (Ai <= 109)

接下來一行m個整數，描述序列B (Bi <= 109)

輸出

輸出一個數表示答案

樣例輸入

7 4
6 6 8 5 5 7 4
7 7 9 6

樣例輸出

2

思路

KMP模板題

AC

#include<iostream> #include<bits/stdc++.h> #define N 1000006 using namespace std; int a[N], b[N], Next[N]; void get_next(int m) { //求Next數組 Next[0] = -1; int i = 0, j = -1;while (i < m) {if(j == -1 || b[i] == b[j]) {Next[++i] = ++j; //賦值 }else {j = Next[j]; //回溯 }} } int kmp(int n, int m) {get_next(m); //求Next數組 int i = 0, j = 0;int ans = 0;while (i < n) {if (j == -1 || a[i] == b[j]) { //當前匹配成功進行下一個匹配 i++;j++; }else {j = Next[j];}if (j == m) { //匹配成功 ans++;j = Next[j]; //進行下一次匹配 }}return ans; }int main() { // freopen("in.txt", "r", stdin);int n, m;while (scanf("%d%d", &n, &m) != EOF) {for (int i = 0; i < n; i++) {scanf("%d",&a[i]);}for (int i = 0; i < m; i++) {scanf("%d", &b[i]);}if (m == 1) { //m = 1 特判 printf("%d\n", n);continue;}for (int i = 1; i < n; i++) {a[i - 1] = a[i] - a[i - 1];}for (int i = 1; i < m; i++) {b[i - 1] = b[i] - b[i - 1];}printf("%d\n", kmp(n - 1, m - 1)); }return 0; }

總結

以上是生活随笔為你收集整理的2264: sequence（KMP）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。