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

题意

字符串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;
}