gym 101667 : H Rock Paper Scissors (FFT 求字符串最大相似度)

7 篇文章 1 订阅
5 篇文章 0 订阅

题目大意:两个人在玩石头剪刀布游戏,你知道另外一个人长度为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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值