CodeForces 300C Beautiful Numbers（排列组合 乘法逆元）

题意：

给出a，b，n三个数

一串由a，b组成的长度为n的数，各个数位相加和为sum，若sum也是由a，b组成的，那么ans++

求最终的ans值

 

思路：

0~n枚举过去，暴力

若此时的sum满足条件，那么那就算出这个条件下的组合数，加到ans上去

c( n,m ) = f[ n ]/(f[ m ]*f[ n-m ])

乘法逆元：

当mod = prime时，则（a/b）%mod = a*( b^(mod-2) )     白书P122

 

#include <iostream>
using namespace std;
#define mod 1000000007
__int64 f[1000050];
int a,b,n;
bool good(__int64 sum)
{
	while(sum)
	{
		if((sum%10)!=a && (sum%10)!=b) return 0;
		sum/=10;
	}
	return 1;
}
__int64 mul(__int64 x,__int64 y)
{
	__int64 ans=1;
	while(y)
	{
		if(y&1) ans = (x*ans)%mod;
		x = (x*x)%mod;
		y>>=1;
	}
	return ans%mod;
}
int main()
{
	f[0] = 1;
	for(int i=1;i<=1000010;i++)
		f[i] = (i*f[i-1])%mod;
	while(cin>>a>>b>>n)
	{
		__int64 ans=0;
		for(int i=0;i<=n;i++)
		{
			__int64 sum = a*i+b*(n-i);
			if(good(sum)==1) 
			{
				__int64 s = (f[i]*f[n-i])%mod;
				ans = (ans+((f[n]*mul(s,mod-2))%mod))%mod;
			}
		}
		cout<<ans<<endl;
	}
	return 0;
}