1. 程式人生 > >51nod 1518 穩定多米諾覆蓋(容斥+二項式反演+狀壓dp)

51nod 1518 穩定多米諾覆蓋(容斥+二項式反演+狀壓dp)

com code clu cto 傳送門 覆蓋 long long 處理 nod

[傳送門[(http://www.51nod.com/Challenge/Problem.html#!#problemId=1518)

解題思路

  直接算不好算,考慮容斥,但並不能把行和列一起加進去容斥,這會使時間復雜度非常高,那麽就考慮枚舉行後\(dp\)。設\(f[i]\)表示存在\(i\)列有線,任意一行無線的方案數,\(g[i[\)表示至少有\(i\)列有線,任意一行無線的方案數,那麽
\[g[i]=\sum\limits_{k=i}^n C(i,k)f[i]\]
二項式反演得
\[f[0]=\sum\limits_{k=0}^n(-1)^kg[k]C(k,0)\]
那麽只需要考慮求出\(g\)
  首先要預處理出來\(dp[i][j]\)

表示\(i\)\(j\)列任意放的方案數,那麽算答案時可以先枚舉哪幾列有線,然後算出\(g[i]\)\(g[i]\)就是首先把\(dp\)數組合並,直觀理解就是把那幾塊拼在一起,然後減去\(j<i\)\(g[j]\),就是保證行沒有線,之後就可以算答案了。時間復雜度O(\(2^n n^2)\)

代碼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#define int long long

using namespace std;
const int N=18;
const int MOD=1e9+7;
typedef long long LL;

int f[N][N],n,m,g[N],ans,dp[2][(1<<(17))],tmp[N];
vector<int> v;

inline void DP(int lim){
    memset(dp,0,sizeof(dp));
    int now=0; dp[0][(1<<lim)-1]=1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=lim;j++){
            now^=1; memset(dp[now],0,sizeof(dp[now]));
            for(int S=0;S<(1<<lim);S++)if(dp[now^1][S]){
                if(S&1) (dp[now][S>>1]+=dp[now^1][S])%=MOD;
                if(i!=1 && (!(S&1))) 
                    (dp[now][(S>>1)|(1<<(lim-1))]+=dp[now^1][S])%=MOD;
                if(j!=1 && (S&1) && lim>1 && (!(S&(1<<(lim-1))))) 
                    (dp[now][(S>>1)|(1<<(lim-1))|(1<<(lim-2))]+=dp[now^1][S])%=MOD; 
            }
        }
        f[i][lim]=dp[now][(1<<lim)-1];
    }
}

inline void prework(){
    for(int i=1;i<=m;i++) DP(i);
}

signed main(){
    n=m=16; prework();
    while(~scanf("%lld%lld",&n,&m)){
        for(int S=(1<<(m-1));S<(1<<m);S++){
            int lst=0; v.clear();
            for(int i=1;i<=m;i++)
                if(S&(1<<(i-1))) v.push_back(i-lst),lst=i;
            for(int i=1;i<=n;i++){
                tmp[i]=1;
                for(int j=0;j<v.size();j++)
                    tmp[i]=1ll*tmp[i]*f[i][v[j]]%MOD;
            }   
            for(int i=1;i<=n;i++){
                g[i]=tmp[i];
                for(int k=1;k<i;k++)
                    g[i]-=1ll*g[k]*tmp[i-k]%MOD,g[i]%=MOD;
            }
            if(v.size()&1) (ans+=g[n])%=MOD;
            else (ans-=g[n])%=MOD;
        } ans=(ans%MOD+MOD)%MOD;
        printf("%lld\n",ans); ans=0;
    }
    return 0;   
}

51nod 1518 穩定多米諾覆蓋(容斥+二項式反演+狀壓dp)