【JZOJ A组】神奇的字符串

Description

Input

Output

Sample Input

9 5 6 4 3
101
11
Query 0
Query 1
Query 2
Query 3
Query 4
Query 5
Query 6
Query 7
Query 8
Change 1
Query 3

Sample Output

0
3
0
2
2
0
3
0
3
1

Data Constraint

思路

首先，因为(N,A)=1，所以0…n-1肯定是一段连续的数
考虑如果两个串的第?位要相同的话，那么我们可以找到满足要求的开头区间。如果从串?取出的串的开头在这个区间里面，那么这两个串的第?为相同，否则不相同，所以题目转化为支持询问一个点被多少个区间包含，并且区间可以改变。
要用动态开点

代码

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int M=100077;
int n,a,b,p,m,c[M],tot=1,Q,x;
char ch[10];
struct node
{
	int l,r,v,lazy;
}tr[10000077];
struct E
{
	int l,r;
};
E work(int x,int y)
{
    int l,r;
    if(y==0) l=p,r=n-1; else l=0,r=p-1;
    l=((l-a*x)%n+n)%n,r=((r-a*x)%n+n)%n;
    return (E){l,r};
}
void Lazy(int x)
{
    tr[tr[x].l].lazy+=tr[x].lazy,tr[tr[x].l].v+=tr[x].lazy;
    tr[tr[x].r].lazy+=tr[x].lazy,tr[tr[x].r].v+=tr[x].lazy;
    tr[x].lazy=0;
}
void change(int d,int l,int r,int L,int R,int k)
{
    if(l==L&&r==R)
	{
		tr[d].v+=k,tr[d].lazy+=k; return;
	}
    int mid=l+r>>1;
    if(!tr[d].l) tr[d].l=++tot; if(!tr[d].r) tr[d].r=++tot;
    Lazy(d);
    if(R<=mid) change(tr[d].l,l,mid,L,R,k); else if(L>mid) change(tr[d].r,mid+1,r,L,R,k);
    else change(tr[d].l,l,mid,L,mid,k),change(tr[d].r,mid+1,r,mid+1,R,k);
}
int Query(int d,int l,int r,int k)
{
    if(l==r) return tr[d].v;
    int mid=l+r>>1;
    if(!tr[d].l) tr[d].l=++tot; if(!tr[d].r) tr[d].r=++tot;
    Lazy(d);
    return (k<=mid)?Query(tr[d].l,l,mid,k):Query(tr[d].r,mid+1,r,k);
}
int main()
{
    scanf("%d%d%d%d%d",&n,&a,&b,&p,&m);
    for(int i=1; i<=m; i++)
    {
        char ch=' '; while(ch!='0'&&ch!='1') ch=getchar();
        c[i]=ch-48; E e=work(i-1,c[i]); 
        if(e.l<=e.r) change(1,0,n-1,e.l,e.r,1); else change(1,0,n-1,0,e.r,1),change(1,0,n-1,e.l,n-1,1);
    }
    scanf("%d\n",&Q);
    while(Q--)
    {
        scanf("%s%d\n",ch,&x);
        if(ch[0]=='Q') printf("%d\n",Query(1,0,n-1,(x*a+b)%n));
        else
        {
            x++; E e=work(x-1,c[x]);
            if(e.l<=e.r) change(1,0,n-1,e.l,e.r,-1); else change(1,0,n-1,0,e.r,-1),change(1,0,n-1,e.l,n-1,-1);
            c[x]=(c[x]+1)%2,e=work(x-1,c[x]);
            if(e.l<=e.r) change(1,0,n-1,e.l,e.r,1); else change(1,0,n-1,0,e.r,1),change(1,0,n-1,e.l,n-1,1);
        }
    }
}