NOIP 模拟 $84\; \rm 猪国杀$

题解 \(by\;zj\varphi\)

设答案为 \(ans\),那么转化问题为求 \(ans\times A^n\)。

发现一定是选最小的最优,所以有 \(60pts\) 可以是枚举最小值。

先放式子

\[ans\times A^n=\sum_{i=0}^{n}\sum_{j=1}^{A}\sum_{k=1}^{n-i}g_{i,j-1,m-j\times k}\binom{n}{i}\sum_{t\ge k}\binom{n-i}{t}(A-j)^{n-i-t} \]

其中 \(g_{i,j,k}\) 表示选了 \(i\) 个最大为 \(j\) 的数,总和不超过 \(k\) 的方案数,之后就是枚举选了多少个 \(j\) 剩下的全要大于 \(j\),随便放。

考虑如何求 \(g\),容斥。

所有式子合并后为

\[ans\times A^n=\sum_{i=0}^{n}\sum_{j=1}^{A}\sum_{k=1}^{n-i}(\sum_{t=0}^{i}(-1)^t\binom{i}{t}\binom{m-k\times j-t\times (j-1)}{i})\binom{n}{i}\sum_{t\ge k}\binom{n-i}{t}(A-j)^{n-i-t} \]

中间这一堆的意思是保证至少有 \(t\) 个大于等于 \(j\) ,然后进行容斥。

\(\binom{m-k\times j-t\times (j-1)}{i}\) 的意思就是把 \(m-k\times j-t\times (j-1)\) 拆成 \(i+1\) 个正整数,然后再给其中的 \(t\) 个加上 \(j-1\) 保证它不小于 \(j\),多拆一个数是为了让 \(i\) 个数的和不用正好为 \(m-k\times j-t\times (j-1)\)。

Code
#include<bits/stdc++.h>
#define ri signed
#define pd(i) ++i
#define bq(i) --i
#define func(x) std::function<x>
namespace IO{
    char buf[1<<21],*p1=buf,*p2=buf;
    #define gc() p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?(-1):*p1++
    #define debug1(x) std::cerr << #x"=" << x << ' '
    #define debug2(x) std::cerr << #x"=" << x << std::endl
    #define Debug(x) assert(x)
    struct nanfeng_stream{
        template<typename T>inline nanfeng_stream &operator>>(T &x) {
            bool f=false;x=0;char ch=gc();
            while(!isdigit(ch)) f|=ch=='-',ch=gc();
            while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=gc();
            return x=f?-x:x,*this;
        }
    }cin;
}
using IO::cin;
namespace nanfeng{
    #define FI FILE *IN
    #define FO FILE *OUT
    template<typename T>inline T cmax(T x,T y) {return x>y?x:y;}
    template<typename T>inline T cmin(T x,T y) {return x>y?y:x;}
    using ll=long long;
    static const int N=1e3+7,MOD=998244353;
    int C[N][N],inv[N],n,m,A;
    ll ans;
    auto fpow=[](int x,int y) {
        int res=1;
        while(y) {
            if (y&1) res=1ll*res*x%MOD;
            x=1ll*x*x%MOD;
            y>>=1;
        }
        return res;
    };
    inline int main() {
        FI=freopen("legend.in","r",stdin);
        FO=freopen("legend.out","w",stdout);
        cin >> n >> m >> A;
        inv[1]=1;
        for (ri i(2);i<=A;pd(i)) inv[i]=1ll*(MOD-MOD/i)*inv[MOD%i]%MOD;
        C[0][0]=1;
        for (ri i(1);i<=1000;pd(i)) {
            C[i][0]=1;
            for (ri j(1);j<=i;pd(j)) C[i][j]=(C[i-1][j-1]+C[i-1][j])%MOD;
        }
        for (ri i(0);i<=n;pd(i))
            for (ri j(1);j<=A;pd(j))
                for (ri k(1);k<=n-i;pd(k)) {
                    ll tmp=0,tq=0,tp,f=1;
                    for (ri t(0),kt;t<=i;pd(t),f=-f) {
                        if ((kt=m-k*j-t*(j-1))<i) break;
                        tmp+=f*C[i][t]*C[kt][i]%MOD;
                    }
                    ((tmp%=MOD)+=MOD)%=MOD;
                    tmp=tmp*C[n][i]%MOD;
                    tp=fpow(A-j,n-i-k);
                    for (ri t(k);t<=n-i;pd(t)) {
                        if (t==n-i) tp=1;
                        tq+=1ll*C[n-i][t]*tp%MOD;
                        (tp*=inv[A-j])%=MOD;
                    }
                    tq%=MOD;
                    (ans+=tmp*tq%MOD)%=MOD;
                }
        printf("%lld\n",ans*fpow(fpow(A,n),MOD-2)%MOD);
        return 0;
    }
}
int main() {return nanfeng::main();}
上一篇:模拟84 考试总结


下一篇:《算法零基础100例》(第84例) 模拟 - 2