2019年天梯赛-全国总决赛-L2-029 特立独行的幸福（25 分)

原创于 2019-03-31 18:41:16 发布 · 2k 阅读

5 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于判断指定范围内哪些数既是快乐数又是素数，并输出这些数及其相关属性。快乐数是指通过迭代过程，将数的每一位平方求和直至得到1的数。该算法还实现了对快乐数序列长度的特别处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/1111914599412858886

#include <iostream>
#include <algorithm>
#include <set>
using namespace std;

int sum(int a){
    int sum=0;
    while(a){
        int t=a%10;
        a/=10;
        sum+=t*t;
    }
    return sum;
}

int prime(int a){
    int i;
    for(i=2; i<a; i++){
        if(a%i==0)
            return 0;
    }
    return 1;
}

int main(){
    int a[1000005];
    int l,r;
    set<int> ss;
    cin >> l >> r;
    for(int i=l;i<=r;i++){
        if(a[i]!=-1){
            int t=i;
            while(t!=1){
                t=sum(t);
                a[t]=-1;
                if(ss.find(t)!=ss.end()){
                    ss.clear();
                    break;
                }
                ss.insert(t);
            }
            a[i]=ss.size();
            ss.clear();
        }
    }
    int flag=1;
    for(int i=l;i<=r;i++){
        if(a[i]>0){
            flag=0;
            if(prime(i)){
                cout << i << " " << a[i]*2 << "\n";
            }else{
                cout << i << " " << a[i] << "\n";
            }
        }
    }
    if(flag){
        cout << "SAD\n";
    }
    return 0;
}