PTA L2-029 特立独行的幸福

对一个十进制数的各位数字做一次平方和，称作一次迭代。如果一个十进制数能通过若干次迭代得到 1，就称该数为幸福数。1 是一个幸福数。此外，例如 19 经过 1 次迭代得到 82，2 次迭代后得到 68，3 次迭代后得到 100，最后得到 1。则 19 就是幸福数。显然，在一个幸福数迭代到 1 的过程中经过的数字都是幸福数，它们的幸福是依附于初始数字的。例如 82、68、100 的幸福是依附于 19 的。而一个特立独行的幸福数，是在一个有限的区间内不依附于任何其它数字的；其独立性就是依附于它的的幸福数的个数。如果这个数还是个素数，则其独立性加倍。例如 19 在区间[1, 100] 内就是一个特立独行的幸福数，其独立性为 2×4=8。

另一方面，如果一个大于1的数字经过数次迭代后进入了死循环，那这个数就不幸福。例如 29 迭代得到 85、89、145、42、20、4、16、37、58、89、…… 可见 89 到 58 形成了死循环，所以 29 就不幸福。

本题就要求你编写程序，列出给定区间内的所有特立独行的幸福数和它的独立性。

输入格式：

输入在第一行给出闭区间的两个端点：1<A<B≤104。

输出格式：

按递增顺序列出给定闭区间 [A,B] 内的所有特立独行的幸福数和它的独立性。每对数字占一行，数字间以 1 个空格分隔。

如果区间内没有幸福数，则在一行中输出 SAD。

输入样例 1：

10 40

输出样例 1：

19 8
23 6
28 3
31 4
32 3

注意：样例中，10、13 也都是幸福数，但它们分别依附于其他数字（如 23、31 等等），所以不输出。其它数字虽然其实也依附于其它幸福数，但因为那些数字不在给定区间 [10, 40] 内，所以它们在给定区间内是特立独行的幸福数。

输入样例 2：

110 120

输出样例 2：

SAD

做法： 

注：如果把每个数都看成一个节点，那么 特立独行的幸福数 必定与1相连且入度为0；

1.确定区间

2.查找每一个点，并记录信息

3.输出

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 10010;

bool visit[N];//是否访问过
int degree[N],of[N];//degree为每个点的入度，of是每个点的平方和
int dist[N];//到 1 的距离，-1则不是幸福数

int get(int n)//平方和
{
    int res = 0;
    while(n)
    {
        int t = n % 10;
        res += t * t;
        n /= 10;
    }
    return res;
}
int prime(int n)//素数 2 否则 1
{
    if(n < 2) return 1;
    for(int i = 2;i <= n / i;i++)
        if(n % i == 0)
            return 1;
    return 2;
}

int find(int x)
{
    if(!visit[x])
    {
        degree[of[x]]++;//后继点的入度++
        visit[x] = true;//已访问
        dist[x] = find(of[x]) + 1;
    }
    //碰到了环
    if(x != 1 && visit[x] && dist[x] == 0) dist[x] = -1;
    return dist[x];
}
int main()
{
    int l = 0,r = 0;
    visit[1] = true;//初始化
    for(int i = 1;i <= 10000;i++) of[i] = get(i);
    
    scanf("%d%d",&l,&r);
    for(int i = l;i <= r;i++)//查找每一个点
        if(!visit[i]) find(i);
    
    int flag = 1;//输出
    for(int i = l;i <= r;i++)
        if(dist[i] != -1 && !degree[i])//可以到达1，且入度为0
        {
            flag = 0;
            printf("%d %lld\n",i,(long long)dist[i] * prime(i));
        }
    
    if(flag) puts("SAD");
    return 0;
}

二刷代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 10010;

int visit[N],st[N],gen[N];

int sq(int x)//平方和
{
    int res = 0;
    while(x)
    {
        int t = x % 10;
        res += t * t;
        x /= 10;
    }
    return res;
}

int prime(int x)
{
    if(x < 2) return 1;
    for(int i = 2;i <= x / i;i++)
        if(x % i == 0) return 1;
    return 2;
}
int dfs(int x)
{
    if(!gen[x] && !visit[x])
    {
        visit[x] = 1;//判断是否有环，和gen数组结合用，可判断是否为幸福数
        int f = sq(x);
        st[f] = 1;//判断特立独行的幸福数用
        gen[x] = dfs(f) + 1;
    }

    if(gen[x]) return gen[x];//gen > 0，则为幸福数
    else return -1;//特殊处理非幸福数
}

int main()
{
    int l = 0,r = 0;
    scanf("%d%d",&l,&r);
    gen[1] = 1;//先设置为1，为判断非幸福数做准备
    for(int i = l;i <= r;i++) dfs(i);
    int cnt = 0;
    for(int i = l;i <= r;i++)
    {
        if(gen[i] && !st[i])//是幸福数，且不依附于其他幸福数
        {
            gen[i]--;//减一
            printf("%d %d\n",i,gen[i] * prime(i));
            cnt++;
        }
    }
    if(!cnt) puts("SAD");
    return 0;
}

结果：