uoj#275. 【清华集训2016】组合数问题（数位dp）

传送门

假设有\(k|{n\choose m}\)，因为\(n!\)中质因子\(k\)的次数为\(S(n)=\left\lfloor\frac{n}{k}\right\rfloor+\left\lfloor\frac{n}{k^2}\right\rfloor+...\)，而\(m!\)和\((n-m)!\)同理。所以如果\(S(n)>S(m)+S(n-m)\)，那么\(k|{n\choose m}\)

不难发现，对于每一个\(k^i\)，\(\left\lfloor\frac{n}{k^i}\right\rfloor\geq \left\lfloor\frac{m}{k^i}\right\rfloor+\left\lfloor\frac{n-m}{k^i}\right\rfloor\)。所以只要有一个\(i\)使得\(\left\lfloor\frac{n}{k^i}\right\rfloor>\left\lfloor\frac{m}{k^i}\right\rfloor+\left\lfloor\frac{n-m}{k^i}\right\rfloor\)，那么\(S(n)>S(m)+S(n-m)\)

当\(i=1\)的时候，设\(n=ak+b,m=ck+d\)，则\(n-m=(a-c)k+b-d\)，那么如果\(b-d<0\)，\(\left\lfloor\frac{n}{k}\right\rfloor=a,\left\lfloor\frac{m}{k}\right\rfloor+\left\lfloor\frac{n-m}{k}\right\rfloor=c+(a-c-1)=a-1<a\)，那么就有\(S(n)>S(m)+S(n-m)\)

同理可得，若\(n\)和\(m\)的\(k\)进制表示下第\(i\)位满足\(n_i<m_i\)，那么就有\(S(n)>S(m)+S(n-m)\)

于是现在的问题就是变成了求\(i<n,j<m\)且\(i\)的\(k\)进制表示下有某一位数值比\(j\)小，可以数位dp，这个就不讲了

然后上面的情况下我们没有考虑\(j>i\)的情况，因为如果\(j>i\)那么\(k\)进制下\(i\)肯定有某一位小于\(j\)，所以我们对于每个\(i\)算出它会多算的个数，发现就是一个等差数列求和的形式，带公式就好了

//minamoto

#include<bits/stdc++.h>

#define R register

#define ll long long

#define fp(i,a,b) for(R int i=a,I=b+1;i<I;++i)

#define fd(i,a,b) for(R int i=a,I=b-1;i>I;--i)

#define go(u) for(int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)

using namespace std;

char buf[1<<21],*p1=buf,*p2=buf;

inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}

ll read(){

    R ll res,f=1;R char ch;

    while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1);

    for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0');

    return res*f;

}

char sr[1<<21],z[20];int C=-1,Z=0;

inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}

void print(R int x){

    if(C>1<<20)Ot();if(x<0)sr[++C]='-',x=-x;

    while(z[++Z]=x%10+48,x/=10);

    while(sr[++C]=z[Z],--Z);sr[++C]='\n';

}

const int N=65,P=1e9+7,inv=500000004;

inline int add(R int x,R int y){return x+y>=P?x+y-P:x+y;}

inline int dec(R int x,R int y){return x-y<0?x-y+P:x-y;}

inline int mul(R int x,R int y){return 1ll*x*y-1ll*x*y/P*P;}

int ksm(R int x,R int y){

	R int res=1;

	for(;y;y>>=1,x=mul(x,x))if(y&1)res=mul(res,x);

	return res;

}

int l1,l2,num1[N],num2[N],k,p,ans;

int f[2][2][2][N];ll n,m;

int dfs(int p,int q,int ok,int pos){

	if(!pos)return ok;

	if(~f[p][q][ok][pos])return f[p][q][ok][pos];

	int res=0,lm1=p?k-1:num1[pos],lm2=q?k-1:num2[pos];

	fp(i,0,lm1)fp(j,0,lm2)

			res=add(res,dfs(p|(i<lm1),q|(j<lm2),ok|(i<j),pos-1));

	return f[p][q][ok][pos]=res;

}

void solve(){

	n=read(),m=read(),l1=l2=0,m=min(n,m),p=m%P;

	memset(f,-1,sizeof(f));

	while(n)num1[++l1]=n%k,n/=k;

	while(m)num2[++l2]=m%k,m/=k;

	while(l2<=l1)num2[++l2]=0;

	ans=dfs(0,0,0,l1);

	ans=dec(ans,1ll*(p+1)*p%P*inv%P);

	printf("%d\n",ans);

}

int main(){

//	freopen("testdata.in","r",stdin);

	int T=read();k=read();

	while(T--)solve();

	return 0;

}

巴特西

uoj#275. 【清华集训2016】组合数问题（数位dp）

最新文章

热门文章