好难的题目啊,是cantalon数的一般情况啊,公式为c(a+b,b)-c(a+b,b-1),一开始想从cantalon推这一题,白浪费了一个小时,又是打了m,n的表和组合数的表,才推了这公式
写完提交后无限WA,直到练习赛时间过了才想起来是我求的com(a+b, b)-com(a+b,b-1)可能为负值,因为a>b不代表a%p>b%p,即(a-b)%p = ((a%p-b%p)+p)%p而不是a%p-b%p,贴个临时的代码,好像代码有点冗长,再修改修改。(已修改)
time:1718ms
//calc(n,k)=n!/(k!*(n-k)!)
//n! =2^a1*3^a2*5^a3……
//k! =2^b1*3^b2*5^b3……
//(n-k)!=2^c1*3^c2*5^c3……
#include<iostream>
#include<string>
using namespace std;
const long long maxn=2000000+1000;
const long long p=20100501;
int cnt=0;
bool not_prime[maxn];
int prime[maxn/10],a[maxn/10],b[maxn/10],c[maxn/10];
void init()//求素数表
{
memset(prime,0,sizeof(prime));
memset(not_prime,0,sizeof(prime));
for(int i=2;i<maxn;i++)
{
if(!not_prime[i]) prime[cnt++]=i;
for(int j=0;j<cnt&&i*prime[j]<maxn;j++)
{
not_prime[i*prime[j]]=1;
if(!(i%prime[j])) break;
}
}
//cout<<cnt<<endl;
}
inline long long fin_fac(long long n,long long q) //n!中q因子个数
{
long long ans=0;
while(n)
{
ans+=n/q;
n/=q;
}
return ans;
}
inline long long fast_pow(long long a,long long k) //a^k%p
{
long long ans=1,t=a;
while(k)
{
if(k&1) ans=ans*t%p;
t=t*t%p;
k>>=1;
}
return ans;
}
long long com(int n,int k) //calc C(n,k)%p=(n!/(k!*(n-k)!))%p = (2^(a1-b1-c1)*3^(a2-b2-c2)*……)%p
{
long long ans=1;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
for(int i=0;prime[i]<=n;i++)
a[i]=fin_fac(n, prime[i]);
for(int i=0;prime[i]<=k;i++)
b[i]=fin_fac(k,prime[i]);
for(int i=0;prime[i]<=n-k;i++)
c[i]=fin_fac(n-k,prime[i]);
for(int i=0;prime[i]<=n;i++)
ans=ans*fast_pow(prime[i],a[i]-b[i]-c[i])%p;
return ans;
}
int main()
{
init();
long long t;
cin>>t;
while(t--)
{
long long a,b;
cin>>a>>b;
if(a<b)cout<<0<<endl;
else if(b==0) cout<<1<<endl;
else
cout<<((com(a+b, b)-com(a+b,b-1))%p+p)%p<<endl;
}
return 0;
}
删了a,b,c三个数组,其它也有调整了一下,快了一倍
time:812ms
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=2000000+1;
const int p=20100501;
bool not_prime[maxn];
int prime[maxn/10];
int fin_fac(int n,int q)
{
if(q==0||n<q) return 0;
int ans=0;
while(n)
{
n/=q;
ans+=n;
}
return ans;
}
ll fast_pow(ll a,ll k)
{
if(k==1) return a;
ll ans=1,t=a;
while(k)
{
if(k&1) ans=ans*t%p;
t=t*t%p;
k>>=1;
}
return ans;
}
ll com(int n,int k)
{
ll ans=1,tmp;
for(int i=0;prime[i]<=n;i++)
ans=ans*fast_pow(prime[i],fin_fac(n, prime[i])-fin_fac(k,prime[i])-fin_fac(n-k,prime[i]))%p;
return ans;
}
int main()
{
int t,a,b,cnt=0;
for(int i=2;i<maxn;i++)
{
if(!not_prime[i]) prime[cnt++]=i;
for(int j=0;j<cnt&&i*prime[j]<maxn;j++)
{
not_prime[i*prime[j]]=1;
if(!(i%prime[j])) break;
}
}
cin.sync_with_stdio(false);
cin>>t;
while(t--)
{
cin>>a>>b;
cout<<(com(a+b, b)-com(a+b,b-1)+p)%p<<endl;
}
return 0;
}
刚刚发现计算com(a+b,b)与com(a+b,b-1)几乎是重复计算了一次,果断改之(也避免了(a-b)%c小于0的问题!)
直接计算com(a+b,b)-com(a+b,b-1)=(a+b)!(n+1-m)/(m!(n+1)!) 注意!不要计算ans=(a+b)!/(m!(n+1)!),再计算ans*(n+1-m),因为不能保证(a+b)!/(m!(n+1)!)可以整除!
time:421MS
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=2000000+1;
const int p=20100501;
bool not_prime[maxn];
int prime[maxn/10];
int fin_fac(int n,int q)
{
if(q==0||n<q) return 0;
int ans=0;
while(n)
{
n/=q;
ans+=n;
}
return ans;
}
ll fast_pow(ll a,ll k)
{
if(k==1) return a;
ll ans=1,t=a;
while(k)
{
if(k&1) ans=ans*t%p;
t=t*t%p;
k>>=1;
}
return ans;
}
int fac(int n,int k)
{
if(k==0) return 0;
int t=0;
while(n%k==0)
{
n/=k;t++;
}
return t;
}
ll calc(int n,int m)
{
ll ans=1;
for(int i=0;prime[i]<=n+m;i++)
ans=ans*fast_pow(prime[i], fac(n+1-m,prime[i])+fin_fac(n+m, prime[i])-fin_fac(m, prime[i])-fin_fac(n+1, prime[i]))%p;
return ans;
}
int main()
{
int t,a,b,cnt=0;
for(int i=2;i<maxn;i++)
{
if(!not_prime[i]) prime[cnt++]=i;
for(int j=0;j<cnt&&i*prime[j]<maxn;j++)
{
not_prime[i*prime[j]]=1;
if(!(i%prime[j])) break;
}
}
cin.sync_with_stdio(false);
cin>>t;
while(t--)
{
cin>>a>>b;
cout<<calc(a, b)<<endl;
}
return 0;
}
发现有15 ms,32ms过的,若哪位大仙知道,求赐教啊!
刚刚找到的公式的推导:
参考自:http://apps.hi.baidu.com/share/detail/17473477
①:我们设初始在坐标系的原点(0,0),从字符串第一位开始,碰到一个1就向上走,碰到一个0就向右走,那么由n个1、m个0组成的字符串最后必定走到(n,m)点,即满足由n个1、m个0组成的字符串的个数为C(n+m,n) = C(n+m,m) (满足n+m长度内n个长度走1或者m个长度走0)。
②:对于任意前缀中1的个数不少于0的个数的字符串的个数这个条件,可以看成是坐标系中,从(0,0)点走到(m, n)点,并且跟y=x-1这条直线不相交的方案数。又因为(0,0)点关于直线y=x-1的对称点是(1,-1),而从(1,-1)点走到(m, n)点的所有方案一定都会与直线y=x-1相交,对于这些方案,将从(1,-1)点到与y=x-1的第一个交点之间的路径关于y=x-1对称翻转过去,就可以得到所有不满足题意的从(0,0)点走到(m, n)点的方案,于是最终答案就是C(n+m, n)-C(n+m,n+1)。