HDOJ 5745 La Vie en rose (bitset优化DP)

最新推荐文章于 2024-02-12 21:28:44 发布

Winterfell30

最新推荐文章于 2024-02-12 21:28:44 发布

阅读量509

点赞数

CC 4.0 BY-SA版权

分类专栏： HDOJ 递推DP 位运算文章标签： dp

本文链接：https://blog.youkuaiyun.com/Dreamon3/article/details/52904871

HDOJ 同时被 3 个专栏收录

22 篇文章

订阅专栏

递推DP

6 篇文章

订阅专栏

位运算

2 篇文章

订阅专栏

本文介绍了一种使用bitset优化的动态规划算法解决字符串匹配问题的方法。该算法允许在一个字符串中查找另一个可以进行一次相邻字符交换的模式字符串。通过巧妙地利用bitset减少空间消耗并加速匹配过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意

字符串s和p，p可以交换相邻两个字符的位置但是只能交换一次。求每个s[i]到s[i+len(p)-1]是否能由p变化得到。

思路

dp[i][j][0/1/2]分别表示在s的i位置和p的j位置时p[j]与p[j-1]交换、不交换、与p[j+1]交换时是否能变换得到。
因为dp是一个bool数组所以我们可以用bitset表示第一维也就是母串的匹配情况，第二维滚动一下。对于s[i]和p[j]的匹配情况也用bitset维护一下，为了防止爆内存这里不需要用p[j]而是表示出26个字母和s[i]的对应情况就可以了。

代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <bitset>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define LL long long
#define Lowbit(x) ((x)&(-x))
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1|1
#define MP(a, b) make_pair(a, b)
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
const int maxn = 1e5 + 7;
const double eps = 1e-8;
const double PI = acos(-1.0);
typedef pair<int, int> pii;

int n, m;
char s[maxn], p[5010];
bitset<maxn> dp[2][3];
bitset<maxn> mp[30];

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int T;
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d%d", &n, &m);
        scanf("%s%s", &s, &p);
        for (int i = 0; i < 30; i++) mp[i].reset();
        for (int i = 0; i < 2; i++)
            for (int j = 0; j < 3; j++)
                dp[i][j].reset();
        for (int i = 0; i < n; i++) mp[s[i]-'a'][i] = 1;
        dp[0][1] = mp[p[0]-'a'];
        if (m > 1) dp[0][2] = mp[p[1]-'a'];
        int now = 0;
        for (int i = 1; i < m; i++)
        {
            now ^= 1;
            dp[now][0] = (dp[now^1][2] << 1) & mp[p[i-1]-'a'];
            dp[now][1] = ((dp[now^1][1] | dp[now^1][0]) << 1) & mp[p[i]-'a'];
            if (i != m - 1) dp[now][2] = ((dp[now^1][0] | dp[now^1][1]) << 1) & mp[p[i+1]-'a'];
        }
        for (int i = 0; i < n; i++)
        {
            if (dp[now][1][i+m-1] || dp[now][0][i+m-1])
                printf("1");
            else
                printf("0");
        }
        printf("\n");
    }
    return 0;
}