CodeChef Counting D-sets (容斥原理+组合数学)

vjudge题面传送门:https://cn.vjudge.net/problem/CodeChef-CNTDSETS

(PS:vjudge上中文版的题面有误,一个点集的直径应该定义为其中点对的切比雪夫距离的最大值。切比雪夫距离是两个点各个维度之差的绝对值取max。这一点看回英文版题面就能知道)


题目分析:一道思维难度较大,代码量极少的题。

直径=d的点集数 = 直径<=d的点集数 - 直径<=d-1的点集数。

考虑如何求直径<=d的点集数。它相当于在一个n维,每一维坐标为0~d的空间内选取点集。为了避免平移导致算重的情况,我们要求这个点集的每一维都要有至少一个0。

如果不看最后的条件,那么答案便是2(d+1)n。根据容斥原理的常见方法,考虑转化成补集,即至少有一维没有0的点集数。由于点集数只跟限制的维数有关,直接用组合数加速子集反演即可。最终答案等于

2(d+1)n+i=1n(1)iCni2di(d+1)ni

注意在指数上进行快速幂时,模数是109+6。再注意一下边界情况(比如d=0等)即可。时间复杂度O(Tnlog(d))


CODE:

#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<stdio.h>
#include<algorithm>
using namespace std;

const int maxn=1010;
const long long M=1000000007;
typedef long long LL;

LL C[maxn][maxn];
int t,n,d;
LL ans;

void Preparation()
{
    for (int i=0; i<maxn; i++) C[i][0]=1;
    for (int i=1; i<maxn; i++)
        for (int j=1; j<=i; j++)
            C[i][j]=(C[i-1][j-1]+C[i-1][j])%M;
}

LL Pow(LL x,LL y,LL z)
{
    if (!y) return 1LL;
    LL tp=Pow(x,y>>1,z);
    tp=tp*tp%z;
    if (y&1) tp=tp*x%z;
    return tp;
}

LL Calc(LL lim)
{
    if (!lim) return 1LL;
    LL ret=Pow(2LL, Pow(lim+1,n,M-1LL) ,M);
    LL sum=0;
    for (int i=1; i<=n; i++)
    {
        LL tp=C[n][i];
        tp=(tp*Pow(2LL, Pow(lim,i,M-1LL)*Pow(lim+1,n-i,M-1LL)%(M-1LL) ,M))%M;
        if (i%2==0) tp=(M-tp)%M;
        sum=(sum+tp)%M;
    }
    ret=(ret-sum+M)%M;
    return ret;
}

int main()
{
    freopen("c.in","r",stdin);
    freopen("c.out","w",stdout);

    Preparation();
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d%d",&n,&d);
        ans=Calc(d);
        if (d) ans=(ans-Calc(d-1)+M)%M;
        printf("%I64d\n",ans);
    }

    return 0;
}
发布了160 篇原创文章 · 获赞 76 · 访问量 10万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览