标签：int res ll 容斥 include 装箱 dp mod

题目链接
参考博客

题目:

• 有n个宝石和n个箱子，每个箱子只能放一个宝石且第i个宝石不能放在a[i]箱子中，问合适的放法数量，mod 998244353;

题解：

总数减去不合法排列的数量就是要的答案。 计算不合法排列数量时容易明白需要用到容斥的做法。得到公式: \(res = n! - \sum_{i = 1}^{n} (-1)^{i - 1} * f[i]\) 其中f[i]是有i个箱子放宝石不合法的排列量。所以问题变为计算f[i];

num[i]表示有num[i]个宝石不能放在i号箱子。dp[i][j]表示前i个箱子中j个箱子不合法的数量。那么有 \(f[i] = (n-i)! * dp[n][i]\)，问题变成计算dp。

dp的转移： \(dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * num[j]\)
最后用容斥的式子计算就可以了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#include<queue>
#include<vector>
#include<string>
#include<fstream>
using namespace std;
#define rep(i, a, n) for(int i = a; i <= n; ++ i);
#define per(i, a, n) for(int i = n; i >= a; -- i);
typedef long long ll;
const int N = 1e4 + 105;
const ll mod = 998244353;
const double Pi = acos(- 1.0);
const int INF = 0x3f3f3f3f;
const int G = 3, Gi = 332748118;
ll qpow(ll a, ll b) { ll res = 1; while(b){ if(b & 1) res = (res * a) % mod; a = (a * a) % mod; b >>= 1;} return res; }
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
//

ll n;
ll a[N], num[N], dp[N], fac[N];


int main()
{
    fac[0] = 1;
    scanf("%lld",&n);
    for(int i = 1; i <= n; ++ i) {
        scanf("%lld",&a[i]);
        num[a[i]] ++;
    }
    for(int i = 1; i <= n; ++ i) fac[i] = 1ll * i * fac[i - 1] % mod;
    dp[0] = 1;
    for(int i = 1; i <= n; ++ i){
        if(!num[i]) continue;
        for(int j = n; j >= 1; -- j){
            dp[j] = (dp[j] + dp[j - 1] * num[i] % mod) % mod;
        }
    }
    
    ll res = fac[n];
    for(int i = 1; i <= n; ++ i){
        if(i & 1) res = (res - fac[n - i] * dp[i] % mod + mod) % mod;
        else res = (res + fac[n - i] * dp[i] % mod) % mod;
    }
    printf("%lld", res);
    return 0;
}

标签：int,res,ll,容斥,include,装箱,dp,mod
来源： https://www.cnblogs.com/A-sc/p/12941191.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。