CodeForces - 834E 暴力枚举+数位dp

定义f(n)为数字n的十进制表示的所有非0数码组成的多重集,求   L<=n<=R 中不同的f(n)的数量。

1e18内出现的所有f(n)一共最多C(27,9)种,暴力枚举每一种,然后check一下是否能组成在(L,R)区间内的数。

这时候用到了数位dp。如下

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll A,B;
int a[20],b[20],num,ans,c[10];
int dfs(int pos,int l,int r)
{
    if(pos<1) return 1;
    if(!l&&!r) return 1;
    int L=l?a[pos]:0;
    int R=r?b[pos]:9;
    for(int i=L;i<=R;i++)
    {
        int flag=0;
        if(c[i])
        {
            c[i]--;
            flag=dfs(pos-1,l&&i==L,r&&i==R);
            c[i]++;
        }
        if(flag) return 1;
    }
}
void solve(int nu,int k)
{
    if(k==9)
    {
        c[k]=nu;
        ans+=dfs(num,1,1);
        return ;
    }
    for(int i=0;i<=nu;i++)
    {
        c[k]=i;
        solve(nu-i,k+1);
    }
}
int main()
{
	while(~scanf("%lld%lld",&A,&B))
	{
	    num=ans=0;
            while(B>0)
            {
                num++;
                a[num]=A%10;
                b[num]=B%10;
                A/=10;
                B/=10;
            }
            memset(c,0,sizeof c);
            solve(num,0);
            printf("%d\n",ans);
	}
	return 0;
}


猜你喜欢

转载自blog.csdn.net/dllpxfire/article/details/80626353
今日推荐