HDU_6078 Wavel Sequence

最新推荐文章于 2017-08-09 18:36:15 发布

长颜草

最新推荐文章于 2017-08-09 18:36:15 发布

阅读量250

点赞数 2

分类专栏： ACM_集训 ACM_动态规划(DP)

本文链接：https://blog.csdn.net/qq_37412229/article/details/76688741

版权

ACM_集训同时被 2 个专栏收录

36 篇文章 0 订阅

订阅专栏

ACM_动态规划(DP)

8 篇文章 0 订阅

订阅专栏

题目链接

题目意思

给你两个序列a，b，让你找出两个函数 f 和 g 使得 a[f]=b[g]，并且a[f1]，a[f2]，a[f3]……a[fk]满足序列a1 < a2 > a3 < a4……为你满足关系的 f 和 g有多少种

解题思路

用两个二维数组sum和dp来存储b数组中每个数字作为波峰和波谷的两种情况：
dp[ ][0]用来表示当前数字为波谷时的情况
dp[ ][1]用来表示当前数字为波峰时的情况
每轮查找都不断更新dp的值
用sum[ ][0]来存储该数字在波谷时的总的情况数
sum[ ][1]用来存储该数字在波峰时的总的情况数
不多说了，具体的看代码吧

代码部分

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
const LL MOD = 998244353;
const int N = 2e3+5;

int b[N], a[N];
LL sum[N][2], dp[N][2];///sum存储b数组中每位数字两种情况下的总值；

int main()
{
    int T, n, m;
    scanf("%d", &T);
    while(T--)
    {
        memset(sum,0,sizeof(sum));
        memset(dp,0,sizeof(dp));
        scanf("%d%d", &n, &m);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        for(int i = 1; i <= m; i++)
            scanf("%d", &b[i]);
        LL ans = 0;
        for(int i = 1; i <= n; i++)
        {
            LL cnt0 = 1, cnt1 = 0;///cnt0表示波谷位置，cnt1表示波峰位置
            for(int j = 1; j <= m; j++)
            {
                dp[j][0] = dp[j][1] = 0;
                if(a[i] == b[j])
                {
                    dp[j][0] = cnt0;///dp[][0]表示b[j]在波谷的情况
                    dp[j][1] = cnt1;///dp[][1]表示b[j]在波峰的情况
                    ans = (ans+cnt1+cnt0)%MOD;///ans表示计算两种情况的总合
                }
                else if(b[j] < a[i])
                    (cnt1 += sum[j][0]) %= MOD;
                else
                    (cnt0 += sum[j][1]) %= MOD;
            }
            for(int j = 1; j <= m; j++)
            {
                if(b[j] == a[i])
                {
                    (sum[j][0] += dp[j][0]) %= MOD;
                    (sum[j][1] += dp[j][1]) %= MOD;
                }
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}