题目
鸽子数字由以下过程定义:从任何正整数开始,将数字替换为其各个数位的平方和,并重复该过程,直到该数字等于1。如果不能,则这个数字不是鸽子数。
例如7是鸽子数,因为7->49->97->130->10->1。(7*7=49,4*4+9*9=97,9*9+7*7=130....如此类推)
显然1是第一个鸽子数。
有Q个询问,每个询问给出一个数k,你需要输出第k个鸽子数。
Input
第一行一个Q,代表询问的个数(Q<=100000)
接下来Q行,每行一个数字k(k<150000)
Output
每行输出一个数,代表第k个鸽子数
Sample Input
2 1 2
Sample Output
1 7
思路:事先预处理下,第150000个鸽子数是1044373, 所以对1~1050000进行判断就好
#include <iostream>
#include <cstdio>
using namespace std;
const int Max = 1050000;
int dove[Max];
int flag[Max];
int ans[150005];
int square_sum(int num)
{
int sum = 0;
while(num) {
int a = num % 10;
sum = sum + a * a;
num /= 10;
}
return sum;
}
bool dove_num(int num)
{
if(flag[num]) {
if(dove[num])
return true;
else
return false;
}
flag[num] = 1;
if(dove_num(square_sum(num)))
dove[num] = 1;
}
int main()
{
int n;
int cnt = 0;
flag[1] = 1;
dove[1] = 1;
for(n = 2; n < Max; n++)
dove_num(n);
for(int i = 1; i < Max; i++)
if(dove[i])
ans[cnt++] = i;
int q, cmd;
// printf("%d\n", cnt);
scanf("%d", &q);
for(int i = 0; i < q; i++) {
scanf("%d", &cmd);
printf("%d\n", ans[cmd-1]);
}
return 0;
}