题目链接
题意:给出两个长度相同的全排列,然后给出一个长度为2 * n的序列,只能对两个去全排列进行取头数字的操作,问构成第二个序列有多少个方案。
题解:dp,dp[i][j]表示匹配第i个元素,在第j个全排列中取,这样我们可以用记忆化的方式将问题求解,补了题之后才明白了dp原来可以有这么多求法,下面是AC代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
#define int long long
const int N=1e6+10,mod=998244353;
int n;
int p[N],q[N],s[N];
int ans;
int dp[N][3];//dp[i][j]表示匹配第i个数,匹配的状态是j
int dfs(int pr,int qr,int tar)//pr是p的指针,qr是q的指针,tar表示的是状态
{
if(pr>n&&qr>n) return 1;
if(dp[pr+qr-1][tar]!=-1) return dp[pr+qr-1][tar];//记忆化
int ans=0;//就是不断的为dp数组赋值
if(p[pr]==s[pr+qr-1] && pr<=n) ans=(ans%mod+dfs(pr+1,qr,0)%mod)%mod;
if(q[qr]==s[pr+qr-1] && qr<=n) ans=(ans%mod+dfs(pr,qr+1,1)%mod)%mod;
return dp[pr+qr-1][tar]=ans;//记忆化
}
signed main()
{
cin.tie(false);
cout.tie(false);
ios::sync_with_stdio(false);
int t;
cin>>t;
while(t--)
{
cin>>n;
for(int i=1;i<=n;i++) cin>>p[i];
for(int i=1;i<=n;i++) cin>>q[i];
for(int i=1;i<=2*n;i++) cin>>s[i];
for(int i=1;i<=2*n+1;i++) dp[i][0]=dp[i][1]=-1;
int ans=0;
if(p[1]==s[1]) ans+=dfs(2,1,0),ans%=mod;
if(q[1]==s[1]) ans+=dfs(1,2,1),ans%=mod;
cout<<ans<<endl;
}
return 0;
}