hdu 4779 Tower Defense(dp+组合计数)

题意:给出n×m的区域,现在有p个重塔和q个轻塔,现在要选出一些塔(至少一个)放入矩形区域,问有多少种方案。其中,两个塔如果在同一行或同一列会相互攻击,但是重塔可以承受一次攻击,但是轻塔不能承受攻击。

思路:这题切入点还是在重塔上,首先,重塔如果放在同一行,那么这一行最多只能放两个重塔,并且,这两个重塔所在的列就不能放其他塔了,一列放两个重塔也同理,所以,可以枚举一下一行放两个重塔和一列放两个重塔的情况,那么,在某一行放了两个重塔,相当于删了两列一行,这样,就可以算出还剩多少行多少列可以放,接下来枚举一下轻塔的放置数量,同理,最后还是算出还有多少行多少列可以放,接下来就放剩下的重塔就行了,这个重塔和轻塔的放置方案是一样的,这个可以预处理出来,然后就可以算了。。。可能是我写的太渣了,擦着边过了,稳居榜尾Orz……


代码:

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<bitset>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-6
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=200+5;
const int mod=1e9+7;
ll dp[maxn][maxn][maxn],C[maxn][maxn],A[maxn][maxn],pinv[maxn];
ll pow_mod(ll x,ll n)
{
    ll res=1;
    while(n)
    {
        if(n&1) res=res*x%mod;
        x=x*x%mod;
        n>>=1;
    }
    return res;
}
ll inv(ll x)
{
    return pow_mod(x,mod-2)%mod;
}
void Init()
{
    memset(C,0,sizeof(C));
    C[0][0]=1;
    for(int i=1;i<maxn;++i)
    {
        C[i][0]=1;
        for(int j=1;j<maxn;++j)
        {
            C[i][j]=C[i-1][j-1]+C[i-1][j];
            if(C[i][j]>=mod) C[i][j]-=mod;
        }
    }
    memset(A,0,sizeof(A));
    A[0][0]=1;
    for(int i=1;i<maxn;++i)
    {
        A[i][0]=1;
        for(int j=i;j>=1;--j)
            A[i][j]=A[i-1][j-1]*i%mod;
    }
    ll x=1;
    pinv[0]=inv(x);
    for(int i=1;i<maxn;++i)
    {
        x=(x<<1)%mod;
        pinv[i]=inv(x);
    }
    memset(dp,0,sizeof(dp));
    for(int i=0;i<maxn;++i)
        for(int j=0;j<maxn;++j)
            dp[i][j][0]=1;
    for(int k=1;k<maxn;++k)
        for(int i=1;i<maxn;++i)
            for(int j=1;j<maxn;++j)
                dp[i][j][k]=(dp[i][j][k-1]+C[i][k]*A[j][k])%mod;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    Init();
    int t,n,m,p,q;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d%d",&n,&m,&p,&q);
        ll ans=0,tmp,tmp2;
        for(int x=0;x<=q;++x)
            for(int i=0;i*2<=p;++i)
                for(int j=0;i*2+j*2<=p;++j)
                {
                    int N=n-i-j*2-x;
                    int M=m-i*2-j-x;
                    if(N<0||M<0) continue;
                    tmp2=C[n][x]*A[m][x]%mod;
                    int tot=min(N*M,p-i*2-j*2);
                    tmp=C[n-x][i]*A[m-x][i*2]%mod*pinv[i]%mod;
                    tmp=tmp*C[m-i*2-x][j]%mod*A[n-i-x][j*2]%mod*pinv[j]%mod;
                    if(i||j||x)
                        ans=(ans+dp[N][M][tot]*tmp2%mod*tmp)%mod;
                    else ans=((ans+dp[N][M][tot]-1)*tmp2%mod*tmp)%mod;
                }
        printf("%I64d\n",ans);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值