BZOJ4377 POI2015 Kurs szybkiego czytania

---

Description

给定n,a,b,p，其中n,a互质。定义一个长度为n的01串c[0..n-1]，其中c[i]==0当且仅当(ai+b) mod n < p。
给定一个长为m的小01串，求出小串在大串中出现了几次。

Input

• Line 1：n,a,b,p,m(2<=n<=10^9,1<=p,a,b,m< n,1<=m<=10^6)。n和a互质。
• Line 2：一个长度为m的01串。

Output

一个整数，表示小串在大串中出现了几次。

Sample Input

9 5 6 4 3
101

Sample Output

3

丫的POI数据辣么大搞得我都不想暴力……

---

$Solution$：

朴素的暴力算法是枚举每个起点，然后$O(m)$判断结果是否符合给定的01串，总时间复杂度$O((n-m)\times m)$$\approx O(n\times m)$。

我们观察题目，发现题目有一句非常随意但是最为关键的话：“其中n,a互质”。怎么理解这句话的意思？

由于a与n互质，所以对于题目中的c序列一定是[0,n-1]的一个全排列。

• 简单证明：假设对于$\forall \; i<j<n，c[i]\%n=c[j]\%n$，则$c[j]-c[i] \equiv 0 \;(mod \; n)$，由于$c[x]=a\times x+b$，所以$c[j]-c[i]=a*(j-i)$，显然因为a与n互质，且$j-i<n$，所以就算$gcd(j-i,n)\neq1$，也一定无法满足$c[j]-c[i] \equiv 0 \;(mod \; n)$。

又由于只要对于当前匹配的这段子串中，只要有一个点不满足题意，这条子串就不能计算在最后答案内，所以我们可以得到以下算法：

• 设该子串开头的数为$x\;(mod\;n)$，则接下来的序列依次为$x+a,x+2a,...\;(mod\;n)$。我们结合给出的01子串，可以得到m组有关$x$的方程。合并这些区间就能得到最后的结果。

根据上述的全排列性质，解出来的$x$的解集依次对应着一个必然唯一确定的子序列。所以该算法正确。

接下来就是如何实现的问题了：直接合并合法区间发现非常困难，因为这些合法区间有些是交集，有些是并集。对于该问题的处理方法参照Codeforces_Round_#376_Div2_problem.D。

#include <bits/stdc++.h>
using namespace std;
static const int M=1000005,N=3000005;
char str[M];
#define Pair pair<int,int>
#define fi first
#define se second
Pair Ans[N];int top=0;
int n,a,b,p,m;
void push(int a,int b,int c,int d){//valid:[a,b)+[c,d)
    if(a)Ans[++top]=make_pair(0,a-1);//[0,a-1]
    if(b<=c-1)Ans[++top]=make_pair(b,c-1);//[b,c-1]
    if(d<=n-1)Ans[++top]=make_pair(d,n-1);//[d,n-1]
}
int main(){
    scanf("%d %d %d %d %d",&n,&a,&b,&p,&m);
    scanf("%s",str);
    Ans[++top]=make_pair(n,n);
    for(int i=0,res=0;i<m;i++,(res+=a)%=n)
        if(str[i]=='0')    //valid:[p-i*a%n,n-i*a%n) or [0,n-i*a%n)+[p-i*a%n+n,n-1)
            push(0,max(0,p-res),n-res,min(n,p-res+n));
        else if(str[i]=='1')     //valid:[0,p-i*a%n)+[n-i*a%n,n-1) or [n-i*a%n,p-i*a%n+n)
            push(max(0,p-res),n-res,min(p-res+n,n),n);
        //c[n-1]=(a+(n-1)*b)%n=(a-b)%n
    int val=b%n;
    for(int i=1;i<m;i++){
        val=(val-a+n)%n;
        Ans[++top]=make_pair(val,val);
    }
    sort(Ans+1,Ans+top+1);
    int ans=0,pre=-1;
    for(int i=1;i<=top;i++){
        if(pre+1<Ans[i].fi)
            ans+=(Ans[i].fi-1)-(pre+1)+1;
        pre=max(pre,Ans[i].se);
    }
    printf("%d\n",ans);
}