F - Two Exam(AtCoder Beginner Contest 238)

原创已于 2022-02-05 23:00:27 修改 · 282 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

计数dp

于 2022-02-05 22:52:54 首次发布

这篇博客探讨了一种基于PPP属性升序选择QQQ属性最小的人员的方法。通过动态规划DP[i][j][k]表示前iii个人中选择jjj个，其中最小未选者的QQQ属性为kkk的方案数。博主详细阐述了状态转移方程，并提供了一个C++实现的示例代码，用于计算特定条件下方案总数。最后，通过遍历得到最终答案并输出。

开发板推荐：天空星STM32F407VET6开发板

超高性价比 STM32主控 | 超高主频 | 一板兼容百芯 | 比赛神器 | 沉金彩色丝印

点击查看

在这里插入图片描述首先按照 $P$ 属性排序所有人，按照 $P$ 属性升序顺序选人，则只需要考虑 $Q$ 属性的大小关系。
设 $D P$ 状态为 $d p [i] [j] [k]$
表示前 $i$ 个人中选择 $j$ 个，其中最小的没有被选择的人的 $Q$ 属性为 $k$ 的方案数量
考虑第 $i + 1$ 个人选或者不选
$如果k>第i个人的Q属性\\dp[i][j+1][k]=(dp[i][j+1][k]+dp[i-1][j][k])\\否则\\dp[i][j][min(k,第i个人的Q属性)]=(dp[i][j][min(k,第i个人的Q属性)]+dp[i-1][j][k])$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
struct Stu
{
    int p,q;
    bool operator <(const Stu &ths)const
    {
        return p<ths.p;
    }
}stu[305];
int dp[305][305][305];//前i个，选了j个，最小的没选的是k的方案数
const int mod=998244353;
int main()
{
    int n,K;
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++)scanf("%d",&stu[i].p);
    for(int i=1;i<=n;i++)scanf("%d",&stu[i].q);
    sort(stu+1,stu+1+n);
    dp[0][0][n+1]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=i-1;j++)
        {
            for(int k=1;k<=n+1;k++)
            {
                dp[i][j][min(k,stu[i].q)]=(dp[i][j][min(k,stu[i].q)]+dp[i-1][j][k])%mod;
                if(k>stu[i].q)dp[i][j+1][k]=(dp[i][j+1][k]+dp[i-1][j][k])%mod;
            }

        }
//        for(int j=0;j<=i;j++)
//            for(int k=1;k<=n+1;k++)printf("dp[%d][%d][%d]=%d\n",i,j,k,dp[i][j][k]);
    }
    ll ans=0;
    for(int i=1;i<=301;i++)ans=(ans+dp[n][K][i])%mod;
    printf("%lld",ans);
    return 0;
}