hdu4602 Partition 快速幂

传送门

记录一下……

题意：通过隔板法，可以把一个数n能拆分成2^(n-1)种方式，给定一个k，问n的所有拆分中有几个k，结果模1e9+7

思路：相当于有一排1，捆绑了k个数，然后看其他的数有多少种拆分方式，如果k个数在两边，那么另外的n-k个数就有2^(n-k-1)种拆分方式，总共有2种情况；如果k个数在中间，那么其它n-k个数总共有2^(n-k-2)种排列方式，总共有n-k-1种情况，这样和2^(n-k)+2(n-k-2)*(n-k-1)就是结果，计算时用快速幂，再特判一下k>n、k=n、k=n-1的情况

完整代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstdlib>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;
LL quick_mod(LL a,LL b,LL mod)
{
    LL ans=1;
    a=a%mod;
    while(b)
    {
        if(b&1)
            ans=ans*a%mod;
        b>>=1;
        a=a*a%mod;
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        LL n,k;
        LL ans;
        scanf("%lld%lld",&n,&k);
        if(k>n) ans=0;
        else if(k==n) ans=1;
        else if(k==n-1) ans=2;
        else ans=quick_mod(2LL,n-k,mod)+(n-k-1)*quick_mod(2LL,n-k-2,mod)%mod;
        ans=ans%mod;
        printf("%lld\n",ans);
    }
    return 0;
}