2017年浙江工业大学大学生程序设计迎新赛决赛 F-栗酱的不等式 [KMP]

来源:互联网 发布:套定额软件 编辑:程序博客网 时间:2024/04/30 06:17

题意:给你两组数列a,b,长度分别为n和m,问满足(a[j]+b[0])%k=(a[j+1]+b[1])%k····=(a[j+m-1]+b[m-1])%k的有多少序列。

题解:通过匹配a上升的趋势与b上升的趋势,通过KMP就可以得到解(a的趋势与b是是反的,所以将a取补操作)

AC代码:

#include<stdio.h>#include<math.h>#include<algorithm>#include<string.h>using namespace std;typedef long long ll;ll t,a[200005],b[200005],n,m,k;ll nxt[200005];void get_nxt(ll l){    memset(nxt,0,sizeof(nxt));    nxt[0]=-1;    ll i=1,j=0;    while(i<l)    {        if(j==-1||b[i]==b[j])            nxt[++i]=++j;        else j=nxt[j];    }}int main(){    scanf("%lld",&t);    while(t--)    {        scanf("%lld%lld%lld",&n,&m,&k);        for(ll i=0;i<n;i++)            scanf("%lld",&a[i]);        for(ll i=0;i<m;i++)            scanf("%lld",&b[i]);        n--,m--;        for(ll i=0;i<n;i++)            a[i]=((a[i+1]-a[i])%k+k)%k;        for(ll i=0;i<m;i++)        {            b[i]=((b[i+1]-b[i])%k+k)%k;            b[i]=(k-b[i])%k;        }        get_nxt(n);        ll i=0,j=0,ans=0;        while(i<n)        {            if(j==-1||a[i]==b[j])            {                i++,j++;                if(j==m)                {                    ans++;                    j=nxt[j];                }            }            else j=nxt[j];        }        printf("%lld\n",ans);    }}


阅读全文
0 0
原创粉丝点击