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

代码如下：

#include<bits/stdc++.h>
using namespace std;
int du[100010];//-1为未统计，0为不为幸福数，>=1表示独立性大小
set<int> depend;//记录运算过程中产生的全部中间值
set<int> repeat;//记录一次运算中是否出现死循环，即不为幸福数

//递归解决，返回数i的独立数大小
int dfs(int i){
    //若统计过，则为出口
    if(du[i]!=-1) return du[i];
    //否则去看它的迭代数情况
    int num=0;
    string str=to_string(i);
    for(int j=0;j<str.size();j++){
        num+=pow(str[j]-'0',2);//累加上去
    }
    //如果为1
    if(num==1) {
    	return 1;
	}
    //否则去看看是否出现过
    if(repeat.find(num)!=repeat.end()){
        //出现过，表示num进入死循环了
        du[num]=0;
        return 0;//表示这个数不为幸福数
    }
    //没有在depend中出现过
    repeat.insert(num);
    du[num]=dfs(num);//递归寻找num的独立性
    if(du[num]!=0){
        //进入依赖的集合中
        depend.insert(num);
        return du[num]+1;
    }else{
        //表示不为幸福数
        return du[num];
    }
}

bool su(int num){
    //判断素数
    for(int i=2;i<=sqrt(num)+1;i++){
        if(num%i==0) return false;
    }
    return true;
}

int main(void){
    int b,e;
    fill(du,du+10001,-1);
    du[1]=1;
    cin>>b>>e;
    //统计区间内全部数的独立性
    for(int i=b;i<=e;i++){
        repeat.clear();
        du[i]=dfs(i);
    }
    //没有进入过depend且du[]中值不为0的数就是特立独行的数
    int cnt=0;
    for(int i=b;i<=e;i++){
        if(du[i]!=0&&depend.find(i)==depend.end()){
            cnt++;
            if(su(i)){
                du[i]*=2;//若为素数则值翻倍
            }
            cout<<i<<" "<<du[i]<<endl;
        }
    }
    if(cnt==0) cout<<"SAD";
    return 0;
}