HDU6078 Wavel Sequence

题目链接

题意

​ 对于一个数列$a_1,a_2,a_3,...,a_n$ ，当且仅当 $a_1<a_2>a_3<a_4>a_5<a_6...$ 时，称该数列构成了波浪。给定了两个数列A,B，分别为 $a_1,a_2,a_3,...,a_n$ , $b_1,b_2,b_3,...,b_m$ 。可以构建两个新的数列 $f_1,f_2,f_3,...,f_k$ , $g_1,g_2,g_3,...,g_k$ 。问能构建多少个 $f,g$ 满足 $a_{f_i} = b_{g_i}$ 且 $a_{f_1},a_{f_2},...,a_{f_k}$ 是一个波浪。

分析

​ 考虑枚举A数列中取到位置 $i$ 。B数列中取到了位置 $j$ 。假设已知 $cnt_0$ 表示取 $a_i$ 之前，前一个值比 $a_i$ 高的种数，即以 $a_i$ 为波谷的种数，$cnt_1$ 为以 $a_i$ 为波峰的种数。那么如果 $a_i ==b_j$ ，对答案的贡献为 $cnt_0 + cnt_1$ 。那么新的问题是如何快速求解 $cnt_0$ 和 $cnt_1$ 。设 $sum[j][k]$ 表示 A 数列最多取到 $i-1$ , B 数列以位置 $j$ 结尾，波峰波谷类型为 $k$ 的种类数($k=1$ 表示波峰，$k=0$表示波谷)。那么 $cnt_0 = \sum_{t=1,b_t >a_i}^{j-1}sum[t][1] , cnt_1=\sum_{t=1,b_t<a_i}^{j-1}sum[t][0]$ 。这个统计可以在枚举 $j$ 的过程中进行。对于 $sum$ 数组的维护，显然只有出现可行方案时需要更新，增加的方案为以 $a_i$ 为结尾的方案数，即对应的 $cnt_0,cnt_1$ 。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 2020
#define eps 1e-7
#define lson rt<<1
#define rson rt<<1|1
#define LL long long
const int mod=998244353;
int a[MAXN];
int b[MAXN];
int sum[MAXN][2];
int main(){
    int T,n,m;
    cin>>T;
    while(T--){
        scanf("%d %d",&n,&m);
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;++i)
            scanf("%d",&a[i]);
        for(int i=1;i<=m;++i)
            scanf("%d",&b[i]);
        int ans=0;
        for(int i=1;i<=n;++i){
            int cnt0=1,cnt1=0;
            for(int j=1;j<=m;++j){
                if(a[i]==b[j]){
                    ans=(ans+(cnt0+cnt1)%mod)%mod;
                    sum[j][1]=(sum[j][1]+cnt1)%mod;
                    sum[j][0]=(sum[j][0]+cnt0)%mod;

                }
                else if(b[j]>a[i])
                    cnt0=(cnt0+sum[j][1])%mod;
                else
                    cnt1=(cnt1+sum[j][0])%mod;
            }
        }
        printf("%I64d\n",ans);
    }
}