C - Contest Setting Gym - 101982C dp 补题

题目链接：https://vjudge.net/contest/273260#problem/C

学习了一下别人的思路，首先去重，然后离散化。

dp数组开二维，每一次更新，状态转移方程，dp[ i ][ j ] = （dp[ i-1] [ j ]+dp[ i-1] [ j -1 ]*(ans[ i ])）%mod;

第i种商品第j种（j<=k）的时候,等于第i-1个商品第j种的总方法数加上第i-1个的第j-1个乘以ans[i]，ans数组记录的是这种商品当前有多少种。

AC代码：

#include<iostream>
#include<stack>
#include<iomanip>
#include<stdio.h>
#include<vector>
#include<map>
#include<queue>
#include<algorithm>
using namespace std;
# define inf 0x3f3f3f3f
# define ll long long
# define mod 998244353
const int maxn=1000+100;
ll dp[maxn][maxn];
ll a[maxn];
map<ll,ll>vis;
map<ll,ll>ans;
int main()
{
    ll n,m;
    scanf("%lld%lld",&n,&m);
    int temp;
    for(int i=1; i<=n; i++)
    {
        scanf("%lld",&a[i]);
    }
    int num=0;
    for(int i=1; i<=n; i++)
    {
        if(vis[a[i]]==0)
        {
            vis[a[i]]=++num;
        }
        ans[vis[a[i]]]++;
    }
//    for(int i=1;i<=num;i++){
//        cout<<i<<" "<<ans[i]<<endl;
//    }
    dp[0][0]=1;
    for(int i=1; i<=num; i++)
    {
        for(int j=0; j<=m; j++)
        {
            int temp=dp[i-1][j];
            if(j!=0)
                temp+=dp[i-1][j-1]*ans[i];
            dp[i][j]=(dp[i][j]+temp)%mod;
        }
    }
    printf("%lld\n",dp[num][m]);
    return 0;
}

 

转载于:https://www.cnblogs.com/letlifestop/p/10262801.html