C++——NOIP模拟题——number

number

题目描述

定义了这样一个函数：
    
如 f15,3 = 53+13 = 126，f22,2 = 22+22 = 8。
如果给定 x，k ，请你求出有多少个非负整数 y 满足：x = fy,k - y 。

输入格式

第一行一个整数 T ，表示数据组数。

接下来 T 行，每行两个整数 x，k。

输出格式

输出共 T 行，每行 1 个数，表示该组数据的答案。答案对 1,000,000,007 取模。

样例数据 1

输入

2 

2 2 

2 4
输出

1 

2

备注

【数据规模与约定】

对于 30% 的数据，满足：y≤105。

对于 60% 的数据，满足：T≤10。

对于 100% 的数据，满足：T≤100；0≤X≤109；1≤k≤9。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node{
long long sum;
int time;
}hash[11][453631];
long long a[11][11],num[10][101000];
int tot[10];
inline int r()
{
int i=0,f=1;
char c;
for(c=getchar();c<'0'||c>'9';c=getchar())
if(c=='-') f=-1;
for(;c>='0'&&c<='9';c=getchar())
i=(i<<3)+(i<<1)+(c^'0');
return i*f;
}
inline int find(int k,long long sum)
{
int i=(sum%453631+453631)%453631;
while(hash[k][i].sum!=sum&&hash[k][i].time!=0) ++i;
return i;
}
inline void dfs(int len,int k,long long sum)
{
if(len==4)
{
int ii=find(k,sum);
if(!hash[k][ii].sum) hash[k][ii].sum=sum;
++hash[k][ii].time;
return;
}
for(int i=0;i<=9;++i) dfs(len-1,k,sum+i*(a[i][k-1]-a[10][len]));
}
inline void bfs(int len,int k,long long sum)
{
if(len==-1)
{
num[k][++tot[k]]=sum;
return;
}
for(int i=0;i<=9;++i) bfs(len-1,k,sum+i*(a[i][k-1]-a[10][len]));
}
int main(void)
{
int t,k,ii,sum;
long long x;
t=r();
for(int i=1;i<=10;++i) a[i][0]=1;
for(int j=1;j<=9;++j)
for(int i=1;i<=10;++i)
a[i][j]=a[i][j-1]*i;
for(int i=1;i<=9;++i) dfs(9,i,0);
for(int i=1;i<=9;++i)
{
bfs(4,i,0);
sort(num[i]+1,num[i]+1+tot[i]);
}
while(t--)
{
x=r();
k=r();
sum=0;
ii=1;
for(int i=2;i<=tot[k];++i)
if(num[k][i]!=num[k][i-1])
{
sum+=(i-ii)*hash[k][find(k,x-num[k][ii])].time;
ii=i;
}
sum+=(tot[k]+1-ii)*hash[k][find(k,x-num[k][ii])].time;
printf("%d\n",sum);
}
return 0;
}