Wavel Sequence HDU - 6078 多校#4 dp

参考小坏蛋_千千

定义dp[i][j]：a[i]与b[j]作为共同串结尾的种类数

关键是要弄清每一轮j循环中，b[j]与a[i]不等的时候要往贡献里怎么加东西，而只有在a[i]==b[j]的时候，才会把贡献登记进答案

举个例子

A：1 5 3

B：4 1 1 5 3

在i=2,j=2的时候，b[j]<a[i]，所以我把此轮的b[j]作为贡献登记进波谷，这个贡献就是b[j]作为波谷的匹配数

#include<bits/stdc++.h>
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<string.h>
#include<iostream>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
using namespace std;
#define ll long long
#define pb push_back
#define FOR(a) for(int i=1;i<=a;i++)
const int inf=0x3f3f3f3f;
const int maxn=1e6+9; 
const int mod=998244353;

int n,m;
int a[maxn],b[maxn];
ll dp[maxn][2];		//b[j]与a[i]作为公共序列结尾的情况

int main(){
	int T;scanf("%d",&T);
	while(T--){
		ll ans=0;
		scanf("%d%d",&n,&m);
		memset(dp,0,sizeof dp);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		for(int i=1;i<=m;i++)scanf("%d",&b[i]);
		for(int i=1;i<=n;i++){//枚举a中的元素
			ll cnt1=1;	//之前可作为波峰的数量
			ll cnt0=0;
			for(int j=1;j<=m;j++){
				if(a[i]==b[j]){
					dp[j][0]+=cnt1;
					dp[j][1]+=cnt0;
					ans=(ans+cnt1+cnt0)%mod;
				}else if(b[j]<a[i])cnt0=(cnt0+dp[j][0])%mod;
				else cnt1=(cnt1+dp[j][1])%mod;
			}

		}
		printf("%lld\n",ans);
	}	
}

最近的题目，难度陡升呢（苦笑