x=f(y,k)-y,等式的右边每位是可以独立算的,最后也可以合并。
当y较大的时候,必然会减成负数,而x>=0, 所以y有一个上限。
9 ^ 9 * 9大概是3e10
9 ^ 9 * 10大概也是3e10
9 ^ 9 * 11是4e10不够11位
所以y最多应该是10位
分y的低5位和高5位分别枚举并查询就可以了。
注意x=0的时候,y=0不能作为解,因为题目要求y>0
注意折半枚举的条件,两个部分可以分开计算,也就是两个部分的计算是独立的;查找的时候可以高效的合并。
注意这题map常数太大,会T,而本身值的个数不太多的时候,查找个数可以考虑全部存起来,用sort和upper_bound - lower_bound
以及注意预处理
//折半枚举
//y最多有10位,否则f(y,k)-y<0
//注意y的各位数字对答案的贡献是**独立**的-->折半搜索
//枚举y的后5位数字,用于枚举前5位数字的时候查询。
//预处理出0~99999中所有数各位k次方的和
//用map会T,因为常数太大!!
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
int T,k;
LL x,s1,s2,a;
LL b[100000];
LL ans;
LL g[100010][10];
LL qpow(LL a, LL b)
{
LL res = 1;
for (LL i=1;i<=b;i++)
res = res * a;
return res;
}
int main()
{
for (int j=1;j<=9;j++)
for (int i=0;i<100000;i++)
{
a = i;
s1 = 0;
while (a)
{
LL tmp = a % 10;
a /= 10;
s1 += qpow(tmp, j);
}
g[i][j] = s1;
}
scanf("%d", &T);
for (int t = 1;t<= T;t++)
{
ans = 0;
scanf("%lld%d", &x, &k);
for (int i=0;i<=99999;i++) b[i] = x - g[i][k] + i;
sort(b,b+100000);
for (LL i=0;i<100000;i++)
{
s2 = g[i][k] - i * 100000;
ans += upper_bound(b,b+100000,s2) - lower_bound(b,b+100000,s2);
}
if (x == 0) ans--;
printf("Case #%d: %lld\n", t, ans);
}
}