最大公共子串-fft(Gym-101667H)

题意,给两个串求最长公共子串。
一看到这个题目显然n^2啊。
知道fft这个做法后什么动态规划弱爆了啊。

解法

首先题目给的串是锤子剪刀布,所以要把其中某个串按照锤子剪刀布的规则变换一下,然后就是最大公共子串的问题了。

给第二个串翻转一下,然后对于每个位置的匹配就都会变成卷积的形式。对每种字符计算:原串为该字符的部位变为1,其余部位变为0。例如串RRRSPP,我们就算字母S,那么这个串就变成000100。最后FFT求出来的每一位就对应着一个位置的匹配。
例如:
1000111和
1111匹配会得到
1111123321
选最大的答案为3。表示x ^ 6(4+2,5+1,6+0) 或者 x ^ 7(4+3,5+2,6+1) 的位置匹配会得到最大值。

注意这题第二个短串的开头不能超过第一个串,所以fft之后次数只能取m-1以上的。

代码:

#include <cstdio>
#include <cstring>
#include <utility>
#include <iostream>
#include <map>
#include <queue>
#include <algorithm>
#include <cmath>
#include <string>
#include <vector>
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
#define N 100010
#define M 2000100
const int INF = 0x3f3f3f3f;
const double eps = 1e-5;
const double PI = acos(-1);
#define NOS ios_base::sync_with_stdio(false);cin.tie(NULL);
#define fi first
#define se second
#define L o<<1
#define tl tree[o].l
#define tr tree[o].r
#define tlw tree[o].lw
#define trw tree[o].rw
#define tmw tree[o].mw
#define rep(i, lll, nnn) for(int i = (lll); i <= (nnn); i++)
const int mod = 1e9+7;

struct Complex {
    double x, y;
    Complex(double x=0, double y=0):x(x), y(y){}
    Complex(const Complex & t) {
        x = t.x; y = t.y;
    }
    Complex operator * (const Complex & t) {
        return Complex(x * t.x - y * t.y, x * t.y + y * t.x);
    }
    Complex operator + (const Complex & t) {
        return Complex(x + t.x, y + t.y);
    }
    Complex operator - (const Complex & t) {
        return Complex(x - t.x, y - t.y);
    }
};

const int MAXN = 300010;
int R[MAXN];
void change(Complex y[], int len)
{
    for(int i = 0; i < len; i++) {
        if(i < R[i]) swap(y[i], y[R[i]]);
    }
}

void fft(Complex y[], int len, int on)
{
    change(y, len);
//cout << "ok1" << endl;
    for(int h = 2; h <= len; h <<= 1) {
        Complex wn(cos(-on * 2 * PI / h), sin(-on * 2 * PI / h));
        for(int j = 0; j < len; j += h) {
            Complex w(1, 0);
            for(int k = j; k < j + h / 2; k++) {
                Complex u = y[k];
                Complex t = w * y[k + h / 2];
                y[k] = u + t;
                y[k + h / 2] = u - t;
                w = w * wn;
            }
        }
//cout << h << endl;
    }
//cout << "ok2" << endl;
    if(on == -1) {
        for(int i = 0; i < len; i++) {
            y[i].x /= len;
        }
    }
}

Complex x1[MAXN], x2[MAXN];

int n, m, len, LL;
char s1[MAXN / 2], s2[MAXN / 2];
int sum[MAXN];

void cal(char c)
{//cout <<c << endl;
    for(int i = 0; i < n; i++) x1[i] = Complex(s1[i] == c, 0);
    for(int i = n; i < len; i++) x1[i] = Complex(0, 0);
    for(int i = 0; i < m; i++) x2[i] = Complex(s2[i] == c, 0);
    for(int i = m; i < len; i++) x2[i] = Complex(0, 0);

    fft(x1, len, 1);
//cout << "ok\n";
    fft(x2, len, 1);
    for(int i = 0; i < len; i++) {
        x1[i] = x1[i] * x2[i];
    }
    fft(x1, len, -1);
    for(int i = 0; i < n + m - 1; i++) {
        sum[i] += int(x1[i].x + 0.5);
//cout << sum[i] << endl;
    }

}

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("data.txt", "r", stdin);
    #endif
    NOS;

    scanf("%d%d", &n, &m);
    scanf("%s%s", s1, s2);

    for(int i = 0; i < m; i++) {
        if(s2[i] == 'R') s2[i] = 'S';
        else if(s2[i] == 'S') s2[i] = 'P';
        else s2[i] = 'R';
    }

    reverse(s2, s2 + m);

    len = 1, LL = 0;
    while(len < n * 2 || len < m * 2) len <<= 1, LL++;
    for(int i = 0; i < len; i++) {
        R[i] = (R[i>>1]>>1)|((i&1)<<(LL-1));
    }

    cal('R');
    cal('P');
    cal('S');

    int ans = 0;
    for(int i = m - 1; i < n + m - 1; i++) ans = max(ans, sum[i]);

    printf("%d\n", ans);

    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值