超级无敌简单题 字节跳动-文远知行杯”广东工业大学第十四届程序设计竞赛

Problem Description

通常来说，题面短的题目一般都比较难，所以我要把题面写得很长很长。
通常来说，题面短的题目一般都比较难，所以我要把题面写得很长很长。
通常来说，题面短的题目一般都比较难，所以我要把题面写得很长很长。
鸽子数字由以下过程定义：从任何正整数开始，将数字替换为其各个数位的平方和，并重复该过程，直到该数字等于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

思路：其实通过打表可以发现一个数是鸽子数，那么它的各个数位的平方和也是鸽子数。所以我先找出1-450 的鸽子数，标记并放入vector 数组。然后从451一直往后枚举 ，如果数 i 的各位数的平方和出现过 那么这个数  i 一定是鸽子数，把它记录下来放入数组，只要找到150000个就好  然后O(1) 查询。

代码如下：

#include<bits/stdc++.h>
using namespace std;
int mp[5000];
map<int,int>mm;
 vector<int>vec;
void cha()
{
  for(int i=1;i<=450;i++)
  {
      int a=i;
      memset(mp,0,sizeof(mp));
      bool flag=false;
      while(a!=1)
      {
          int k=a;
          int ans=0;
          while(k)
          {
              int s=k%10;
              ans+=s*s;
              k/=10;
          }
          if(mp[ans]==1)
          {
              flag=true;
              break;
          }
          mp[ans]=1;
          a=ans;
      }
      if(!flag)
      {
          vec.push_back(i);
          mm[i]=1;
      }
  }
}
int main()
{
    cha();
    for(int i=451;i<=1050000;i++)
    {
          int k=i;
          int ans=0;
          while(k)
          {
              int s=k%10;
              ans+=s*s;
              k/=10;
          }
          if(mm[ans]==1)
          {
            vec.push_back(i);
            mm[i]=1;
          }

      }
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int a;
        scanf("%d",&a);
        printf("%d\n",vec[a-1]);
    }
}