BZOJ4377: [POI2015]Kurs szybkiego czytania

学到了求区间交并的姿势..

注意到a,n互质，所以对于所有的0<=i< n，ai+b mod n互不相同，一一对应
我们令s=ai+b，问大串包含多少个小串，就是问有多少个s使得s,s+a,s+2a,s+3a…..满足限制
于是我们可以列出若干个形如l<=s< r(Mod n)的限制
这些不等式解集的交就是能取的s的数量，减掉尾部的m-1个i就是答案（后缀不足m不能作为开头）
卡常卡了好久qwq

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1e9
using namespace std;

const int maxm = 1000010;

int n,A,B,p,m;
char str[maxm];

struct node
{
    int x,c;
    node(){}
    node(const int _x,const int _c){x=_x;c=_c;}
}a[maxm<<2]; int len;
int t[maxm<<1],tp;

inline bool cmp(const node x,const node y){ return x.x<y.x; }

int main()
{
    scanf("%d%d%d%d%d",&n,&A,&B,&p,&m);
    scanf("%s",str);

    for(int i=0,k=0;i<m;i++,k+=A)
    {
        if(k>=n) k-=n;

        int l,r;
        if(str[i]=='0') l=0,r=p-1;
        else l=p,r=n-1;
        l-=k; if(l<0) l+=n;
        r-=k; if(r<0) r+=n;

        if(l<=r)
        {
            r++;
            a[++len]=node(l,1);
            if(r<n) a[++len]=node(r,-1);
        }
        else
        {
            a[++len]=node(0,1);
            a[++len]=node(r+1,-1);
            a[++len]=node(l,1);
        }
    }
    for(int i=n-m+1,k=((ll)A*i+(ll)B)%n;i<n;i++,k+=A)
    {
        if(k>=n) k-=n;
        t[++tp]=k;
    }
    sort(a+1,a+len+1,cmp);
    sort(t+1,t+tp+1);

    int c=0,ans=0;
    int pos=1;
    for(int i=1;i<=len;)
    {
        int j; int tc=c;
        for(j=i;j<=len&&a[j].x==a[i].x;j++) c+=a[j].c;
        int temp=0; while(t[pos]<a[i].x&&pos<=tp) pos++,temp++;
        if(tc==m) ans+=a[i].x-a[i-1].x-temp;
        i=j;
    }
    int temp=0; while(t[pos]<n&&pos<=tp) pos++,temp++;
    if(c==m) ans+=n-a[len].x-temp;
    printf("%d\n",ans);

    return 0;
}