工程课 0. pat
原题链接
简单
作者:
彩
,
2024-04-19 18:02:58
,
所有人可见
,
阅读 2
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cctype>
#include <vector>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int N = 1e4 + 10;
int n, m;
int a[N];
int b[N];
bool st[N];
bool is_prime(int n)
{
if(n < 2) return false;
if(n == 2) return true;
for(int i = 2; i <= n / i; i++)
if(n % i == 0)
return false;
return true;
}
int fun(int x)
{
int sum = 0;
string s = to_string(x);
for(int i = 0; i < s.length(); i++)
{
int t = s[i] - '0';
sum += t * t;
}
return sum;
}
void init(int n, int m)
{
for(int i = n; i <= m; i++)
{
int x = i;
vector<int> v;
bool is_ok = true;
while(x != 1)
{
x = fun(x);
//v.push_back(x);
if(find(v.begin(), v.end(), x) != v.end() && (x >= n && x <= m))
{
is_ok = false;
break;
}
v.push_back(x);
}
if(is_ok)
{
st[i] = true;
b[i] = v.size();
if(is_prime(i)) b[i] *= 2;
for(auto t : v) st[t] = false;
}
else st[i] = false;
}
}
int main()
{
cin >> n >> m;
//for(int i = n; i <= m; i++) st[i] = true;
init(n, m);
int cnt = 0;
for(int i = n; i <= m; i++)
if(st[i])
{
cout << i << " " << b[i] << endl;
cnt++;
}
if(cnt == 0) cout << "SAD" << endl;
return 0;
}