问题重述
学会字符串哈希后,动态规划选手九峰想要出一道解法为字符串哈希题,于是wcy给他口胡了一道题,却把九峰难倒了,你能帮他解决这个问题吗?
给定长度为n的字符串序列a和字符串k,询问a有多少子序列拼接起来等于k。
输入样例
5 abcba
ab c abc ba abcba
输出
3
拼接后等于abcba的子序列有三种:[1,2,4],[3,4],[5]
代码
1.My
#include <vector>
#include <iostream>
#include <string.h>
#include <map>
#include <stack>
#define ll long long
using namespace std;
const int N=5e6+10;
int n,P=131;
int dp[N];
ll h[N],p[N],b[N];
string s;
vector<string> v;
ll Get(int l,int r){
return h[r]-h[l-1]*p[r-l+1];//相当于h[l-1]左移r-(l-1)位,注意是h[l-1]不是h[l]
}
int main()
{
cin>>n>>s;
for(int i=0;i<n;i++){
string t;
cin>>t;
v.push_back(t);
}
p[0]=1;
for(int i=1;i<=s.size();i++){//h[i]记录s从第一个字符->第i个字符之间字符串的hash值
h[i]=h[i-1]*P+s[i-1]-'a';
p[i]=p[i-1]*P;
}
dp[0]=1;//边界表示dp[0][0]用前0个v中的字符串填充长度为0的s子空字符串的组合数为1,恰好满足
//dp[i][j]表示用v中的前i个字符串去装长度为j的s子字符串的组合个数
for(int i=1;i<=n;i++){//每次只用考虑当前字符串的贡献,因为前面的字符串贡献都已经算了
string t=v[i-1];
int len=t.size();
for(int k=1;k<=len;k++){
b[k]=b[k-1]*P+t[k-1]-'a';
}
//for(int j=len;j<=s.size();j++)//!!!!注意遍历顺序要倒着遍历
//因为这一层的dp是由上一层左上方的dp推出来的
//如果正着遍历则原本上一层的dp会被这一层的dp覆盖掉,而我们需要的实际上是上一层的dp
//为了避免被覆盖掉,必须逆着遍历
for(int j=s.size();j>=len;j--){
if(Get(j-len+1,j)==b[len]){//表示后面可以斗上了就看前面的个数了
dp[j]+=dp[j-len];//有点像图轮,u->v,所以到v的路径数就取决于到u的路径数
//如果不能填充j-len,则dp[j]=0+dp[j-len]=0
}
}
}
cout<<dp[s.size()];
system("pause");
return 0;
}
2.others
#include <stdio.h>
#include <vector>
#include <iostream>
#include <string.h>
#define ll long long
using namespace std;
const int N=5000010,P=131,q=1e9+7;
int n;
char s[N],t[N];
ll h[N],p[N],b[N];
int dp[N];
ll Get(int l,int r){
//return ((h[r]-h[l-1]*p[r-l+1])%q+q)%q;
return h[r]-(h[l-1]*p[r-l+1]);
}
int main()
{
scanf("%d",&n);
scanf("%s",s+1);
int l=strlen(s+1);//是字符串开始的下标
p[0]=1;
for(int i=1;i<=l;i++){
p[i]=p[i-1]*P;
//h[i]=(h[i-1]*P+s[i]-'a')%q;
h[i]=h[i-1]*P+s[i]-'a';
}
dp[0]=1;
for(int i=1;i<=n;i++){
scanf("%s",t+1);
int x=strlen(t+1);//
for(int j=1;j<=x;j++){
//b[i]=(b[i]*P+t[j]-'a')%q;
b[i]=b[i]*P+t[j]-'a';
}
for(int j=l-x+1;j>=1;j--){//遍历区间左端点
if(Get(j,j+x-1)==b[i]){
dp[j+x-1]=dp[j+x-1]+dp[j-1];
}
}
}
cout<<dp[l];
system("pause");
return 0;
}