Wavel Sequence HDU

参考小坏蛋_千千

定义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);}}

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