The 2021 Shanghai Collegiate Programming Contest_D. Zztrans 的班级合照

D. Zztrans 的班级合照

DP

题目传送门：

题目传送门

题面：

思路：

一开始在想组合数学，就是去重部分做一下，然后再枚举安排重复的部分。写不出，主要太繁琐了，而且前后不同状态有转移，不是很好弄。
还是要考虑DP。
我觉得我的DP有了很大的进步。

1. $d{p}_{i,j}=$前$i$个人里面，有$j$个去第二排，$i-j$个去第一排。
2. 然后把初始的数组转化为：
   第$i$名有$a_i$个人。
3. 然后对同一排名的人按照同一批次处理。
4. 因为排名一样的人可以随便放，所以最后答案要记得乘以每一个$a_i!$。
5. 把a数组sort一下，保证有序性，这样就可以通过把序号靠后的加到第二排来保证后排更高。（傻卵了，重打了一遍发现干嘛sort）
6. 再对重复的部分枚举一下放置情况，总共$a_i$个人同一排名，$k$在第一排，$a_i-k$在第二排。

循环范围的真谛：
无论哪个阶段，只可能第一排人比第二排多，不可能第二排人更多的。

最终转移式子：
$$d{p}_{sum,j}=\sum d{p}_{sum-a_i,j-k}$$
其中，j应该是选中了sum里的后j个数字。
k应该是从$0$枚举到$min（i，j）$。

代码：

发现自己一个很坏很坏的习惯，喜欢CV板子和CV重复代码，还是要多打打，码力明明很差了。

以后写题解代码都再打一遍。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int maxn = 5e3 + 10;
const int mod = 998244353;
ll a[maxn], vis[maxn], dp[maxn][maxn];
vector<ll> v;
ll f[maxn];

int main() {
    f[0] = 1;
    for (ll i = 1; i <= 5002; i++)
        f[i] = f[i - 1] * i % mod;
    ll n;
    cin >> n;
    for (ll i = 1; i <= n; i++) {
        cin >> a[i];
        vis[a[i]]++;
    }
    for (ll i = 1; i <= n; i++) {
        if (vis[i]) v.push_back(vis[i]);
    }
    ll sum = 0;
    dp[0][0]=1;
    for (auto i:v) {
        sum += i;
        for (ll j = min(n / 2, sum); j >= sum - j; j--) {
        //第二排人小于等于第一排人
            for (ll k = 0ll; k <= min(i, j); k++)
                dp[sum][j] = (dp[sum][j] + dp[sum - i][j - k]) % mod;
        }
    }
    ll ans = dp[n][n / 2];
   for(auto i:v){
        ans = ans * f[i] % mod;
    }
    cout<<ans<<endl;
}