标签：nxt 数列 int 移项 cin 差分 栗酱 kmp fo

 链接：https://ac.nowcoder.com/acm/contest/27589/C
来源：牛客网

题目描述

栗酱有一个长度为n的数列A，一个长度为m的数列B，现在询问A中有多少个长度为m的连续子序列A'，
满足(a'₁+b₁)%k = (a'₂+b₂)%k = …… = (a'_m + b_m)%k。

输入描述:

第一行一个数T，表示有T组数据。 对于每组数据， 第一行三个整数，n, m, k。 第一行输入n个数, a₁,a₂,…,a_n, 表示A数列中的数， 第二行输入m个数, b₁,b₂,…,b_m, 表示B数列中的数。

输出描述:

每一组数据输出一行，满足条件的连续子序列数量。

示例1

输入

复制 2 3 2 5 7 8 7 8 7 3 2 5 7 8 9 8 7

2
3 2 5
7 8 7
8 7
3 2 5
7 8 9
8 7

输出

复制 1 2

1
2

备注:

T≤15, 2≤m≤n≤2×10⁵, 1≤a_i,b_i,k≤10⁹

分析：

%运算是可以满足结合律的

移项就变成了差分运算

只要存在 (a[i] - a[i+1]) % k == -(b[i] - b[i+1) % k 就行了

在a的差分数组里kmp b的差分数组 

//-------------------------代码----------------------------

//#define int ll
const int N = 3e6+10,base = 1e9+10;
int n,m,k;
int a[N],b[N];
int ca[N],cb[N];
int ha[N],hb[N];
int nxt[N];

void solve()
{
//    cin>>n>>m;
    cin>>n>>m>>k;
    fo(i,1,n) a[i] = b[i] = nxt[i] = 0;
    fo(i,1,n) cin>>a[i],a[i] %= k;
    fo(i,1,m) cin>>b[i],b[i] %= k;
    fo(i,1,n-1) a[i] = (a[i+1] - a[i] + k) % k;
    fo(i,1,m-1) b[i] = (b[i] - b[i+1] + k) % k;
    n -- ,m -- ;
    fo(i,2,m) {
        nxt[i] = nxt[i-1];
        while(nxt[i] && b[nxt[i] + 1] != b[i]) nxt[i] = nxt[nxt[i]];
        nxt[i] += (b[nxt[i] + 1] == b[i]);
    }
    fo(i,1,m) {
//         cout<<nxt[i]<<' ';
    }
//     cout<<endl;
    int j = 1,cnt = 0;
    fo(i,1,n) {
        while(j != 1 && b[j] != a[i]) j = nxt[j-1] + 1;
        if(a[i] == b[j]) j ++ ;
        if(j == m + 1) {
            cnt ++ ;
            j = nxt[j-1] + 1;
        }
    }
    cout<<cnt<<endl;
}
void main_init() {}
signed main(){
    AC();clapping();TLE;
    cout<<fixed<<setprecision(12);
    main_init();
//  while(cin>>n,n)
//  while(cin>>n>>m,n,m)
    int t;cin>>t;while(t -- )
    solve();
//    {solve(); }
    return 0;
}

/*样例区


*/

//------------------------------------------------------------

 

标签：nxt,数列,int,移项,cin,差分,栗酱,kmp,fo
来源： https://www.cnblogs.com/er007/p/16686782.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。