Problem H Rock Paper Scissors，FFT

最新推荐文章于 2020-09-30 14:34:58 发布

phython96

最新推荐文章于 2020-09-30 14:34:58 发布

阅读量427

点赞数

分类专栏： ACM-ICPC训练题解快速傅里叶系列

本文链接：https://blog.csdn.net/weixin_37517391/article/details/80154299

版权

ACM-ICPC训练题解同时被 2 个专栏收录

241 篇文章 13 订阅

订阅专栏

快速傅里叶系列

8 篇文章 0 订阅

订阅专栏

本文介绍了一种利用快速傅立叶变换(FFT)算法来解决游戏策略匹配问题的方法。具体而言，通过将石头剪刀布游戏的策略转换为能够直接进行数学运算的形式，并运用FFT算法找到最优匹配策略，从而最大化赢得比赛的概率。

摘要由CSDN通过智能技术生成

题目

题目链接

题意

给出两段石头剪刀布的顺序 $S$ 和 $T$ ，其中 $T$ 要短一些，现在让你把 $T$ 往 $S$ 的某个位置上靠，使得靠好了以后， $T$ 能赢 $S$ 的子段的次数最大。
如图：这里写图片描述

题解

这道题很典型的FFT啦，首先我们把 $T$ 序列换成它能赢的序列 $T'$ ，也就是 $T$ 序列中的 $R、S、P$ 对应的换成 $S、P、R$ 形成 $T'$ 。

这样的话，我们只需要在 $S$ 中找一段匹配程度最大的就可以了，这的最大的匹配程度就是答案。

为了匹配，我们把 $T'$ 倒换过来，记为 $rT'$ ，我们想象一下做卷积的过程，
发现新的卷积序列中的第 $k$ 个位置的值等于 $S[k-lenT,k-1]$ 与 $T'$ 序列对应位置乘积之和。

为了使用卷积解决这个问题，我们把问题拆成3部分，即单独考虑 $P、S、R$ 时候，最大匹配程度，最后将相同位置的匹配程度加起来就可以了。

代码

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
double pi = acos(-1.0);
struct complex{
    double re,im;
    complex(double r = 0.0,double i = 0.0):re(r),im(i){};
    complex operator+(complex com){
        return complex(re+com.re,im+com.im);
    }
    complex operator-(complex com){
        return complex(re-com.re,im-com.im);
    }
    complex operator*(complex com){
        return complex(re*com.re-im*com.im,re*com.im+im*com.re);
    }
};
complex wn,wntmp;
void rader(complex arr[],int n){
    int num = n-1;
    for(int i = 0;i < n;++i){
        int tn = n>>1;
        while(num && num >= tn) num ^= tn,tn >>= 1;
        num |= tn;
        if(num > i) swap(arr[i],arr[num]);
    }
}
void FFT(complex cs[],int n,int f){
    rader(cs,n);
    for(int s = 1;s < n;s <<= 1){
        wn = complex(cos(f*2*pi/(s*2)),sin(f*2*pi/(s*2)));
        for(int offset = 0;offset < n;offset += s<<1){
            wntmp = complex(1.0,0.0);
            for(int i = 0;i < s;++i){
                complex u = cs[offset+i],v = cs[offset+i+s]*wntmp;
                cs[offset+i] = u + v;
                cs[offset+i+s] = u - v;
                wntmp = wntmp * wn;
            }
        }
    }
    if(f == -1)
        for(int i = 0;i < n;++i)
            cs[i].re /= n;
}
int n,m;
const int maxn = 1e5+7;
char S[maxn],T[maxn];
int ans[maxn*4];
complex csA[maxn*4],csB[maxn*4];
#define pr(x) cout<<#x<<":"<<x<<endl
void solve(char c){
    memset(csA,0,sizeof(csA));
    memset(csB,0,sizeof(csB));
    for(int i = 0;i < n;++i) csA[i] = complex(S[i]==c?1.0:0);
    for(int i = 0;i < m;++i) csB[i] = complex(T[i]==c?1.0:0);
    int len = 1;
    while(len < n) len<<=1;
    len <<= 1;
    FFT(csA,len,1);
    FFT(csB,len,1);
    for(int i = 0;i < len;++i) csA[i] = csA[i]*csB[i];
    FFT(csA,len,-1);
    for(int i = m-1;i < len;++i) {
        ans[i] += int(csA[i].re+0.5);
    };

}
char big(char c){
    if(c == 'R') return 'S';
    if(c == 'S') return 'P';
    if(c == 'P') return 'R';
}
int main(){
    cin>>n>>m>>S>>T;
    for(int i = 0;i < m/2;++i) swap(T[i],T[m-1-i]);
    for(int i = 0;i < m;++i) T[i] = big(T[i]);
    solve('P');
    solve('S');
    solve('R');

    int mx = 0;
    for(int i = 0;i < n+m+1;++i) mx = max(mx,ans[i]);
    cout<<mx<<endl;

    return 0;
}