【题目链接】http://acm.hdu.edu.cn/showproblem.php?pid=6078
题目意思
给定两个数组问你同时满足波浪的波浪数。不理解可以看题目提示部分
解题思路
首先从a数组一个个往下查询b数组是否有相同数字。在查询新的数字时候就要判断前面是否有可以与新加的数字组成波浪的,而当新加数比前面的数大时候就可以添加前面数字为波谷的情况下,而自己添加后处在波峰的位子(如前面有的1 3,现在加入5的话就会考虑15和53的情况,这时候1和3处在波谷而5处在波峰
)反之波谷。所以开二维数组dp来存储的处在波峰的情况个数和波谷的个数。。。又因为a数组可能有相同数字出现所以开s二维数组储存之前的状况。具体如代码
代码部分
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll p=998244353;
//const LL MAXN=1e6+10;
int a[2005],b[2005];
ll dp[2][2005],s[2][2005];///dp数组存本轮的值,s数组存储总次数的(防止a数组出现相同数字)
int main()
{
int t;
scanf("%d",&t);
while (t--)
{
int n,m;
ll sum=0;
memset(dp,0,sizeof(dp));
memset(s,0,sizeof(s));
scanf("%d %d",&n,&m);
for (int i=0; i<n; i++)
scanf("%d",&a[i]);
for (int i=0; i<m; i++)
scanf("%d",&b[i]);
for (int i=0; i<n; i++)
{
ll ans1=1; ///统计为前面波谷的情况,同时自己属于波峰
ll ans0=0; ///统计为前面波峰的情况,自己属于波谷
for (int j=0; j<m; j++)
{
if (a[i]==b[j]) ///相等更新对应波峰波谷的值
{
dp[0][j]=ans1; ///波峰
dp[1][j]=ans0; ///波谷
sum=(sum+ans1+ans0)%p; ///总情况数
}
else if(a[i]> b[j]) ///a[i]比b[j]打所以b可以嫁接a处在波峰的情况
ans0+=s[0][j]%=p;
else ans1+=s[1][j]%=p; ///反之嫁接波谷的情况
}
for (int j=0; j<m; j++)
{
if (a[i]==b[j]) ///把之前的a[i]的次数加上本轮的
{
s[0][j]+=dp[0][j]%=p;
s[1][j]+=dp[1][j]%=p;
}
}
}
printf("%lld\n",sum);
}
return 0;
}