2017 计蒜之道 初赛 第一场 B.阿里天池的新任务

 1 /* QYP kuai wo dai ma*/
 2 #include<algorithm>
 3 #include<iostream>
 4 #include<iomanip>
 5 #include<cstring>
 6 #include<cstdlib>
 7 #include<cstdio>
 8 #include<queue>
 9 #include<ctime>
10 #include<cmath>
11 #include<stack>
12 #include<map>
13 #include<set>
14 #define rep(i,a,b) for(register int i=a;i<=b;i++)
15 #define ll long long
16 #define re register
17 using namespace std;
18 const int N=1e6;
19 int w[N+10];
20 int n,a,b,L,R;
21 char s[N+10],t[N+10];
22 int nxt[N+10];
23 inline int gi() {
24     re int res=0;
25     char ch=getchar();
26     while(ch<‘0‘||ch>‘9‘) ch=getchar();
27     while(ch>=‘0‘&&ch<=‘9‘) res=res*10+ch-‘0‘,ch=getchar();
28     return res;
29 }
30 void get_w() {
31     w[1]=b;
32     for(re int i=2;i<=n;i++) w[i]=(w[i-1]+a)%n;
33 }
34 void get_s() {
35     for(re int i=1;i<=n;i++) {
36         if(w[i]>=L&&w[i]<=R) {
37             if(w[i]%2==0) s[i]=‘A‘;
38             else s[i]=‘T‘;
39         }
40         else {
41             if(w[i]%2==0) s[i]=‘G‘;
42             else s[i]=‘C‘;
43         }
44     }
45 }
46 void NEXT() {
47     int len=strlen(t+1);
48     nxt[1]=0;
49     int j=0;
50     for(re int i=2;i<=len;i++) {
51         while(j&&t[j+1]!=t[i]) j=nxt[j];
52         if(t[j+1]==t[i]) nxt[i]=++j;
53     }
54 }
55 void KMP() {
56     int lens=strlen(s+1),j=0;
57     int lent=strlen(t+1);
58     int ans=0;
59     for(re int i=1;i<=lens;++i) {
60         while(j&&t[j+1]!=s[i]) j=nxt[j];
61         if(t[j+1]==s[i]) ++j;
62         if(j==lent) {ans++;j=nxt[j];}
63     }
64     cout<<ans;
65 }
66 int main() {
67     freopen("2.in","r",stdin);
68     freopen("2.out","w",stdout);
69     n=gi(),a=gi(),b=gi(),L=gi(),R=gi();
70     s[0]=‘@‘,t[0]=‘$‘;
71     scanf("%s",t+1);
72     get_w();
73     get_s();
74     NEXT();
75     KMP();
76     return 0;
77 }