去年多校赛的一道题目,今天下午拿来训练的时候做的,刚开始题意卡了半天,根本没有看懂,后来才发现那个下标的大小决定了我们题目理解的是否正确==
总之大意就是,有一串长字符s,短字符p,p中相邻的字母可以进行交换但不能重复交换,问以s的每个字符开头是否能匹配p或者p的变形。
理解题意后发现这道题的思路还是相当简单的,就是一个很普通的dp,但是问题是数据规模,正常的dp时间复杂度为o(m*n),5e9绝对超时(虽然听说当时暴力可过,但是现在数据加强了),通过看别人的题解学到了新姿势,用bieset来优化。
首先因为空间不够所以我们使用滚动数组,空间压缩之后就是时间的优化。
我们注意到我们的dp数组是一个只有01的布尔类型,所以可以将其中一维看成一个二进制数,用bitset来存储。
至于其他细节,在代码注释中。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include <cmath>
#include<iostream>
#include<algorithm>
#include <bitset>
#define N 100005
#define M 5010
#define ll long long
using namespace std;
char s[N],p[M];
bitset<N> dp[2][3]; //即bool dp[2][3][N]
bitset<N> w[30]; //用来存储s中每个字母出现的位置
/*
如果不用bitset优化的情况下,正常的动规方程如下:
(1为不交换,0为与前一个交换,2为与后一个交换
dp[i][j][1]=(dp[i-1][j-1][0]||dp[i-1][j-1][1])&&(s[i]=p[j])
dp[i][j][0]=(dp[i-1][j-1][2])&&(s[i]==p[j-1])
dp[i][j][2]=(dp[i-1][j-1][0]||dp[i-1][j-1][1])&&(s[i]=p[j+1])
*/
int main()
{
int t,m,n;
cin>>t;
while(t--)
{
cin>>n>>m;
cin>>s>>p;
for(int i=0;i<30;i++)
w[i].reset();
for(int i=0;i<n;i++)
w[s[i]-'a'][i]=1;
for(int i=0;i<2;i++)
for(int j=0;j<3;j++)
dp[i][j].reset();
int cur=0; //(dp[i-1][j-1][0]||dp[i-1][j-1][1])&&(s[i]=p[j])滚动数组的状态下标
dp[0][1]=w[p[0]-'a']; //如果不交换可以满足,那么在s中能匹配p[0]这个字符的就是所有出现[0]的地方,即[p[0]-'a']
if(m>1) dp[0][2]=w[p[1]-'a']; //与后面一个交换后可以满足说明一定可以匹配p中第二个字符
for(int j=1;j<m;j++)
{
cur^=1;
dp[cur][0]=(dp[cur^1][2]<<1)&w[p[j-1]-'a']; //这里要左移一位,因为dp[cur^1][]中的每个i与dp[cur][]中的序列是相同的,而我们现在需要的是i-1与之对应,所以需要左移一位同时与上的也是w[p[j-1]-'a']
dp[cur][1]=((dp[cur^1][0]|dp[cur^1][1])<<1)&w[p[j]-'a'];
if(j+1<m) dp[cur][2]=((dp[cur^1][0]|dp[cur^1][1])<<1)&w[p[j+1]-'a'];
}
for(int i=0;i<n;i++)
if(dp[cur][0][i+m-1]||dp[cur][1][i+m-1]) printf("1");
else printf("0");
printf("\n");
}
return 0;
}
这份代码时间也是卡过……第一次T了第二次交就过了,好迷QAQ