HDU 5519 Kykneion asma（沈阳站K题&&DP+容斥）-CSDN博客

本文链接：https://blog.csdn.net/u014610830/article/details/49816237

这题想了好久没想出来，最后是学习的这里的做法解决的：K题题解

基于FFT，本弱到现在还是不会写，所以并不能提供FFT的写法。

分析：设 $W(n,a_0,a_1,a_2,a_3,a_4)$ 代表可以有前导 $0$ 的 $n$ 位 $5$ 进制数字，至多 $a_i$ 个数字 $i$ 的方案数。那么

a n s = W (n - 1, a 0, a 1 - 1, a 2, a 3, a 4) + W (n - 1, a 0, a 1, a 2 - 1, a 3, a 4) + W (n - 1, a 0, a 1, a 2, a 3 - 1, a 4) + W (n - 1, a 0, a 1, a 2, a 3, a 4 - 1)

$ans=\\W(n-1,a_0,a_1-1,a_2,a_3,a_4)\\+W(n-1,a_0,a_1,a_2-1,a_3,a_4)\\+W(n-1,a_0,a_1,a_2,a_3-1,a_4)\\+W(n-1,a_0,a_1,a_2,a_3,a_4-1)$
现在问题变成了如何求

W(n,a0,a1,a2,a3,a4) $W(n,a_0,a_1,a_2,a_3,a_4)$ ，这里用容斥来求：
设

V(n,a0,a1,a2,a3,a4) $V(n,a_0,a_1,a_2,a_3,a_4)$ 代表

n $n$ 位

5 $5$ 进制的数字，

0,1,2,3,4 $0,1,2,3,4$ 中有数字超过规定限制的情况。

W (n, a 0, a 1, a 2, a 3, a 4) = 5 n - V (n, a 0, a 1, a 2, a 3, a 4)

$W(n,a_0,a_1,a_2,a_3,a_4)=5^{n}-V(n,a_0,a_1,a_2,a_3,a_4)$

设 $S(a_{b_1},a_{b_2}..,a_{b_m})$ 代表数 $b_1,b_2,..,b_m$ 一定超过限制的方案数。

V (n, a 0, a 1, a 2, a 3, a 4) = \sum i S (a i) - \sum i, j, i, j 互 不 相 等 S (a i, a j) + \sum i, j, k, i, j, k 互 不 相 等 S (a i, a j, a k) - \sum i, j, k, l, i, j, k, l 互 不 相 等 S (a i, a j, a k, a l) + S (a 0, a 1, a 2, a 3, a 4)

$V(n,a_0,a_1,a_2,a_3,a_4)=\\\sum_{i}S(a_i)-\sum_{i,j,i,j互不相等}S(a_i,a_j)+\sum_{i,j,k,i,j,k互不相等}S(a_i,a_j,a_k)\\-\sum_{i,j,k,l,i,j,k,l互不相等}S(a_i,a_j,a_k,a_l)+S(a_0,a_1,a_2,a_3,a_4)$

$dp[i][msk][MSK]$ 代表最终有状态 $MSK$ 的数一定超过限制，当前 $i$ 个数，有 $msk$ 个数超过限制，这里可以 $msk$ 是 $MSK$ 的子状态。

注意点：这里 $MSK$ 中存在的数， $msk$ 不存在的数，不能填入 $i$ 位。

设 $cnt[msk]$ 代表这个状态下 $1$ 的个数
这样得到

d p [i] [m s k] [M S K] = d p [i - 1] [m s k] [M S K] * (5 - c n t [M S K] + c n t [m s k]) + \sum (m s k $ 1 < < k) > 0 d p [i - a [k] - 1] [m s k | 1 < < k] [M S K] * (i - 1 a [ k ])

$dp[i][msk][MSK]=\\dp[i-1][msk][MSK]*(5-cnt[MSK]+cnt[msk])\\+\sum_{(msk\$1<<k)>0}dp[i-a[k]-1][msk|1<<k][MSK]*{{i-1}\choose{a[k]}}$

这里有一个东西： $MSK$ 这个状态在表达式中只提供了一个 $1$ 的个数，这个时候把 $MSK$ 直接定义成最终有多少个数字超过了限制，最后一维设成 $j$ ：

d p [i] [m s k] [j] = d p [i - 1] [m s k] [j] * (5 - j + c n t [m s k]) + \sum (m s k $ 1 < < k) > 0 d p [i - a [k] - 1] [m s k | 1 < < k] [j] * (i - 1 a [ k ])

$dp[i][msk][j]=dp[i-1][msk][j]*(5-j+cnt[msk])\\+\sum_{(msk\$1<<k)>0}dp[i-a[k]-1][msk|1<<k][j]*{{i-1}\choose{a[k]}}$

此时：

S (a b 1, a b 2 . ., a b m) = \sum c n t [m s k] = j d p [n] [m s k] [j]

$S(a_{b_1},a_{b_2}..,a_{b_m})=\sum_{cnt[msk]=j}{dp[n][msk][j]}$

此时复杂度为 $o(4*n*2^5*5^2)$

附上代码：

#include <bits/stdc++.h>
#include <algorithm>
#define LL long long
#define FOR(i,x,y)  for(int i = x;i < y;++ i)
#define IFOR(i,x,y) for(int i = x;i < y;-- i)

using namespace std;

typedef vector <int> VT;

const int maxn = 15010;
const int maxm = 30030;
const int Mod = 1e9+7;
const int MSK = 1<<5;

int n,a[5];

void gcd(LL a,LL b,LL& d,LL& x,LL& y){
    if(!b)  {d = a;x = 1;y = 0;}
    else{gcd(b,a%b,d,y,x);y -= x*(a/b);}
}

LL Inv(LL a,LL n){
    LL d,x,y;
    gcd(a,n,d,x,y);
    return d == 1 ? (x+n)%n : -1;
}

LL fac[maxn],inv[maxn],cnt[MSK];

void init(){
    fac[0] = 1; inv[0] = 1;
    FOR(i,1,maxn){
        fac[i] = i*fac[i-1]%Mod;
        LL p = Inv(i,Mod);
        inv[i] = inv[i-1]*p%Mod;
    }
    FOR(i,0,MSK)    cnt[i] = __builtin_popcount(i);
}

LL dp[maxn][MSK];
LL Pow;

LL C(LL p,LL q){
    return fac[p]*inv[q]%Mod*inv[p-q]%Mod;
}

LL quickpow(LL a,LL n,LL m){
    LL ans=1;
    while(n){
        if(n&1) ans = (ans*a)%m;
        a = (a*a)%m;
        n>>=1;
    }
    return ans;
}

void work(){
    LL ans = 0;
    FOR(p,1,5){
        if(!a[p])   continue;
        ans += Pow;
        ans %= Mod;
        -- a[p];
        -- n;
        FOR(j,1,6){
            memset(dp,0,sizeof(dp));
            dp[0][0] = 1;
            FOR(i,1,n+1){
                FOR(msk,0,MSK){
                    if(cnt[msk] <= j){
                        dp[i][msk] = dp[i-1][msk]*(5-j+cnt[msk])%Mod;
                        FOR(k,0,5){
                            if((msk & (1<<k)) && a[k] <= i-1){
                                dp[i][msk] += dp[i-a[k]-1][msk^(1<<k)]*C(i-1,a[k]);
                                dp[i][msk] %= Mod;
                            }
                        }
                    }
                }
            }
            FOR(msk,0,MSK){
                if(cnt[msk] == j){
                    if(j % 2){
                        ans -= dp[n][msk];
                        ans = (ans + Mod)%Mod;
                    }
                    else{
                        ans += dp[n][msk];
                        ans %= Mod;
                    }
                }
            }
        }
        ++ n;
        ++ a[p];
    }
    printf("%I64d\n",ans);
}


int main()
{
    //freopen("test.in","r",stdin);
    init();
    int T,tCase = 0;    scanf("%d",&T);
    while(T--){
        printf("Case #%d: ",++tCase);
        scanf("%d",&n);
        FOR(i,0,5)  scanf("%d",&a[i]);
        Pow = quickpow(5,n-1,Mod);
        work();
    }
    return 0;
}