题目大意:两个人在玩石头剪刀布游戏,你知道另外一个人长度为n的策略序列以及你自己的机器的长度为m的策略序列,你可以从任意一个点开始连续出m轮,问你最多能赢几次。
感谢大佬博客的题解和实现
https://blog.csdn.net/acterminate/article/details/79325574
如果将某一个串变一下,就是两个串的匹配,求最大相似度。但两个串的长度都是 1 0 5 10 ^ 5 105,暴力搞不了。
FFT解法:由于只有三种字母,可以分别枚举每个字母,求所有位置开始匹配该字母的贡献,求三遍,将答案相加取最大值。
一个字母从某个位置开始的贡献的求法:对原串和目的串做相应变动,首先将原串 R 变 P,P变S,S变R。对当前处理的字母C,遍历原串和子串,如果该位置为C则该位置的值为1,否则为0,这时每一个起始位置的贡献为对应位置相乘的和,但如果将目的串翻转一下,则贡献变成类似卷积的一个计算式子,FFT可以用来加速卷积的运算,因此可以用FFT快速计算出所有点开始匹配的贡献。
因为两个串不一样长(目的串更短),怎么处理使得两个串的卷积可以得出每一个点开始匹配的贡献?
将目的串翻转后在目的串的结尾添0,这时可以发现卷积的每一项正好是从某个位置开始匹配的贡献,原理是,翻转前原串和目的串要从某一个位置开始匹配,可以加前导0使得目的串原来的起点和匹配点对齐,然后对前缀串匹配即可,这个前缀串翻转过来变成交叉相乘正好是卷积的某一项。
因此只要将目的串翻转后添后导0至和原串一样长,两个串的卷积就能求出所有点开始匹配的贡献。
因为只要求最大相似度,FFT算法过程很多细节不用处理,对答案是不影响的。
代码:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;
const int maxn = 4e5 + 10;
const double pi = acos(-1.0);
const double PI = acos(-1.0);
int n,m;
struct complex {
double r,i;
complex (double _r = 0.0,double _i = 0.0) {
r = _r;
i = _i;
}
complex operator + (const complex & rhs) {
return complex(r + rhs.r,i + rhs.i);
}
complex operator - (const complex & rhs) {
return complex(r - rhs.r,i - rhs.i);
}
complex operator * (const complex & rhs) {
return complex(r * rhs.r - i * rhs.i,r * rhs.i + i * rhs.r);
}
};
char s[maxn],t[maxn];
int ans[maxn],sum[maxn];
complex A[maxn],B[maxn];
void change(complex a[],int len) {
int tot = 0;
while(1 << tot < len) tot++;
tot--;
for(int i = 0; i < len; i++) {
int cur = 0;
for(int j = 0; j <= tot; j++) {
if(i & (1 << j))
cur |= 1 << (tot - j);
}
if(i < cur) {
swap(a[i],a[cur]);
}
}
}
/*
常数更小的change方法
*/
/*void change(complex y[],int len)
{
int i,j,k;
for(i = 1, j = len/2;i < len-1; i++)
{
if(i < j)swap(y[i],y[j]),printf("%d %d\n",i,j);
//交换互为小标反转的元素,i<j保证交换一次
//i做正常的+1,j左反转类型的+1,始终保持i和j是反转的
k = len/2;
while( j >= k)
{
j -= k;
k /= 2;
}
if(j < k) j += k;
}
} */
void FFT(complex a[],int len,int type = 1) {
change(a,len);
for(int i = 2; i <= len; i <<= 1) {
complex w = complex(cos(type * 2 * pi / i),sin(type * 2 * pi / i));
for(int j = 0; j < len; j += i) {
complex wp = complex(1,0);
for(int k = 0; k < i / 2; k++) {
complex t = wp * a[j + k + i / 2];
complex u = a[j + k];
a[j + k] = u + t;
a[j + k + i / 2] = u - t;
wp = wp * w;
}
}
}
if(type == -1) {
for(int i = 0; i < len; i++)
a[i].r /= len;
}
}
void solve(char c) {
int len = 1;
while(len < 2 * n || len < 2 * m) len <<= 1;
for(int i = 0; i < n; i++)
A[i] = complex(s[i] == c,0);
for(int i = n; i < len; i++)
A[i] = complex(0,0);
for(int i = 0; i < m; i++)
B[i] = complex(t[i] == c,0);
for(int i = m; i < len; i++)
B[i] = complex(0,0);
FFT(A,len,1);
FFT(B,len,1);
for(int i = 0; i < len; i++) {
A[i] = A[i] * B[i];
}
FFT(A,len,-1);
for(int i = 0; i < len; i++) {
sum[i] = (int) (A[i].r + 0.5);
}
len = n + m - 1;
for(int i = 0; i < len; i++) {
ans[i] += sum[i];
}
}
int main() {
scanf("%d%d%s%s",&n,&m,s,t);
for(int i = 0; i < n; i++) {
if(s[i] == 'R') s[i] = 'P';
else if(s[i] == 'S') s[i] = 'R';
else if(s[i] == 'P') s[i] = 'S';
}
reverse(t,t + m);
solve('S');
solve('P');
solve('R');
int ANS = 0;
for(int i = m - 1; i < n + m - 1; i++) {
ANS = max(ANS,ans[i]);
}
printf("%d\n",ANS);
return 0;
}