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;
}