C - Contest Setting Gym - 101982C

比较简单的一个dp,但是自己没做出来,看过大佬博客之后秒懂了https://blog.csdn.net/aqa2037299560/article/details/84643172

(做个好事)

他的思路先统计相同难度题目的个数(和离散化差不多),状态转移方程是这样的 dp[i][j] = dp[i - 1][j - 1] * val[i] + dp[i - 1][j];

也就是前i - 1个数挑出j个的方案数(这是之前就求出来的,相当于继承了下来),dp[i - 1][j - 1] * val[i] 这个是新增加的一道题目所带来的方案书的增加量。说的通俗一点就是:现在的 = 之前的 + 新增加的。(引用)

#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int mod = 998244353;
int a[1010];
ll val[1010];
ll dp[1010][1010];

int  main()
{
    int n,k;
    cin >> n >> k;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    sort(a + 1, a + 1 + n);
    int cnt = 1;
    int top = 1;
    for(int i = 2; i <= n; i++)
    {
        if(a[i] == a[i - 1])
        {
            cnt++;
        }
        else
        {
            val[top++] = cnt;
            cnt = 1;
        }
    }
    val[top++] = cnt;
    memset(dp, 0, sizeof(dp));
    for(int i = 0; i <= 1000; i++)
    {
        dp[i][0] = 1;
    }
    for(int i = 1; i <= top - 1; i++)
    {
        for(int j = 1; j <= k; j++)
        {

            dp[i][j] = ((dp[i - 1][j - 1] * val[i] % mod) + dp[i - 1][j]) % mod;
        }
    }
    cout << dp[top - 1][k] << endl;
    return 0;
}

 

你可能感兴趣的:(ACM)