[BZOJ4737][lucas]清华集训2016：组合数问题

BZOJ4737

lucas定理的简单证明：
$C_n^m\%p=C_{n/p}^{m/p}\ast C_{n\%p}^{m\%p}\%p$
首先有$(1+x{)}^p\equiv 1+x^p(\%p)$，由费马小定理易证
又$(1+x{)}^p\equiv (1+x{)}^{\lfloor n/p\rfloor \ast p}(1+x{)}^r$
$\equiv (1+x^p{)}^{\lfloor n/p\rfloor }(1+x{)}^r$
二项式定理：$\equiv {\sum }_{i=0}^k{C}_{\lfloor \frac{n}{p}\rfloor }^i{x}^{pi}{\sum }_{i=0}^k{C}_r^j{x}^j$
左边第m项系数为$C_n^m$
右边第m项系数为$C_{\lfloor \frac{n}{p}\rfloor }^i{C}_r^j(pi+j=m,j<p)$
$pi+j=m$看作除法的形式即可得证

对于本题，我们将i,j看作k进制数，则要求i,j的每一位分别构成的组合数之积模k=0
因为是k进制数，且k为质数，所以这样的组合数不存在k的因子
则必须要求存在某一位，该位上i的数字小于j的数字，这样这一位的组合数就为0
则数位dp即可

Code：

#include<bits/stdc++.h>
#define mod 1000000007
#define ll long long 
using namespace std;
inline ll read(){
	ll res=0,ff=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') ff=-ff;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*ff; 
}
const int N=1e5+5,inv2=5e8+4;
int t,k;
int l1,l2,lim,p1[N],p2[N],c[205][205],f[N][2][2][2];
ll m,n;
inline void add(int &x,int y){x+=y;if(x>=mod) x-=mod;}
int dfs(int flag,int q,int p,int now){
	if(!now) return flag;
	if(f[now][flag][p][q]!=-1) return f[now][flag][p][q];
	int lim1=(q?(k-1):p1[now]),lim2=(p?(k-1):p2[now]);
	int &tmp=f[now][flag][p][q];
	tmp=0;
	for(int i=0;i<=lim1;i++)
		for(int j=0;j<=lim2;j++)
			add(tmp,dfs(flag|(i<j),q|(i<lim1),p|(j<lim2),now-1));
	return f[now][flag][p][q];
}
int main(){
	t=read();k=read();
	while(t--){
		n=read();m=read();
		l1=l2=0;m=min(n,m);lim=m%mod;
		memset(f,-1,sizeof(f));
		while(n) p1[++l1]=(n%k),n/=k;
		while(m) p2[++l2]=(m%k),m/=k;
		while(l2<=l1) p2[++l2]=0;
		int ans=dfs(0,0,0,l1);
		add(ans,mod-(1ll*(lim+1)*lim%mod*inv2%mod));
		cout<<ans<<"\n";
	}
	return 0;
}