CF163A Substring and Subsequence 題解

分析

考慮 DP。

定義狀態函式 $f_{i,j}$ 表示在 $s[1\dots i]$ 中選一個子串 $a$，在 $t[1 \dots j]$ 中選一個子序列 $b$，且 $s_i,t_j$ 必選時 $a=b$ 的方案數。則有兩種情況：

1. $s_i \ne t_j$。此時 $a$ 和 $b$ 是不可能相同的了，所以 $f_{i,j}=0$。
2. $s_i = t_j$。在只選 $s_i,t_j$ 時有 $1$ 種方案，而不止選 $s_i,t_j$ 時根據列舉前一個字元選的位置，有 $\sum\limits_{k=1}^{j-1} f_{i-1,k}$ 種方案。所以 $f_{i,j}=1+\sum\limits_{k=1}^{j-1} f_{i-1,k}$。

整合一下有轉移方程：$f_{i,j}=[s_i=t_j]\times(\sum\limits_{k=1}^{j-1}f_{i-1,k}+1)$。定義 $s_{i,j}=\sum\limits_{k=1}^{j} f_{i,k}$，則有：$f_{i,j}=[s_i=t_j]\times(s_{i,j-1}+1)$。這就是個字首和最佳化，在更新完 $f_{i,j}$ 只後直接更新 $s_{i,j}$ 即可。

答案就是 $\sum\limits_{i=1}^{{|s|}\sum\limits_{j=1}}f_{i,j}$。複雜度 $O(n^2)$。

程式碼

#include<bits/stdc++.h>
using namespace std;
//#define int long long
#define re register
#define il inline
#define pii pair<int,int>
#define x first
#define y second
#define gc getchar()
#define rd read()
#define debug() puts("------------")

namespace yzqwq{
	il int read(){
		int x=0,f=1;char ch=gc;
		while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc;}
		while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^48),ch=gc;
		return x*f;
	}
	il int qmi(int a,int b,int p){
		int ans=1;
		while(b){
			if(b&1) ans=ans*a%p;
			a=a*a%p,b>>=1;
		}
		return ans;
	}
	il auto max(auto a,auto b){return (a>b?a:b);}
	il auto min(auto a,auto b){return (a<b?a:b);}
	il int gcd(int a,int b){
		if(!b) return a;
		return gcd(b,a%b);
	}
	il int lcm(int a,int b){
		return a/gcd(a,b)*b;
	}
	il void exgcd(int a,int b,int &x,int &y){
		if(!b) return x=1,y=0,void(0);
		exgcd(b,a%b,x,y);
		int t=x;
		x=y,y=t-a/b*x;
		return ;
	}
	mt19937 rnd(time(0));
}
using namespace yzqwq;

const int N=5005,p=1e9+7;
int f[N][N],s[N][N];
char a[N],b[N];
int n,m;

il void solve(){
	scanf("%s",a+1),n=strlen(a+1);
	scanf("%s",b+1),m=strlen(b+1);
	for(re int i=1;i<=n;++i)
	for(re int j=1;j<=m;++j){
		f[i][j]=(long long)((a[i]==b[j])*(s[i-1][j-1]+1))%p;
		s[i][j]=(long long)(s[i][j-1]+f[i][j])%p;
	}
	long long ans=0;
	for(re int i=1;i<=n;++i) ans=(ans+s[i][m])%p;
	printf("%lld\n",ans);
	return ;
}

signed main(){
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	int t=1;while(t--)
	solve();
	return 0;
}