HDNOIP201405杨辉三角

2016.1.27

试题描述

杨辉三角是形如如下的数字三角形：

1

1    1

1   2    1

……

现在想求出杨辉三角第N行的N个数中，有多少个数能被给定的质数p整除。

输入

一行两个空格隔开的整数N和p

输出

输出一个整数，表示第N行能被给定的质数p整除的个数

输入示例

3 2

输出示例

1

其他说明

对于60%的数据，N≤30，p≤1000，对于80%的数据，N≤1000，p≤1000，对于100%的数据，N≤〖10〗^9，p≤1000

首先知道是卢卡斯定理 和 组合数取模

然后就一个一个试呗

#include<iostream>
using namespace std;
int a[],e,ans;
int main()
{
    int n,p,b,ct;
    scanf("%d%d",&n,&p);
    b=n-=;
    while(b)
    {
        a[++e]=b%p;
        b/=p;
    }
    for(int i=;i<=n;i++)
    {
        b=i;ct=;
        while(b)
        {
            if(b%p>a[ct]) {ans+=;break;}
            else {ct++;b/=p;}
        }
    }
    printf("%d",ans);
}

然后果断TLE

然后根据杨辉三角的对称性，砍一半

#include<iostream>
using namespace std;
int a[],e,ans;
int main()
{
    int n,p,b,ct;
    scanf("%d%d",&n,&p);
    b=n-=;
    while(b)
    {
        a[++e]=b%p;
        b/=p;
    }
    for(int i=;i<(n+)/;i++)
    {
        b=i;ct=;
        while(b)
        {
            if(b%p>a[ct]) {ans+=;break;}
            else {ct++;b/=p;}
        }
    }
    ans*=;
    if(n+&)
    {
        b=n/;ct=;
        while(b)
        {
            if(b%p>a[ct]) {ans+=;break;}
            else {ct++;b/=p;}
        }
    }
    printf("%d",ans);
}

然并卵，依旧TLE

于是机智的我想到了构造

还想到了状态压缩

就是二进制某一位为1表示构造的数在p进制下该位上比n在对应位上大

#include<iostream>
using namespace std;
int a[],e,ans;
int main()
{
    int n,p,b,ct;
    scanf("%d%d",&n,&p);
    b=n-=;
    while(b)
    {
        a[++e]=b%p;
        b/=p;
    }
    for(int t = e- ; t >=  ; t-- )
    {
        for(int i = (  << t ) -  ; i >=  ; i-- )
        {
            b=i;ct=a[t+];
            for(int j = t- ; j >= ; j-- )
            {
                if(<<j&b) ct*=p--a[j+];
                else ct*=a[j+]+;
            }
            ans+=ct;
        }
    }
    printf("%d",ans);
}

AC后激动的我瞬间觉得我有做神犇的潜质

但我发现其他人的代码都特短。。。

我方了

冷静后，发现我傻*了

明明可以反着算。。。要知道根据卢卡斯定理构造模p不等于0的数有多简单。。。

看了代码瞬间就懂的

AC代码：

#include<iostream>
using namespace std;
int e,ans=;
int main()
{
    int n,p,b;
    scanf("%d%d",&n,&p);
    b=n-;
    while(b)
    {
        ans*=b%p+;
        b/=p;
    }
    printf("%d",n-ans);
}