L2-029 特立独行的幸福 (25 分)

题目链接

题目大意：给一个区间让求出在这个区间内的独立幸福数和该幸福数的值，例如19->82->68->100->1可以达到1，即是幸福数，独立的意思是该数是变换序列的开始，例如这个样例中82就不是独立的，因为82可以由19衍生出来。

#include <bits/stdc++.h>

using namespace std;

bool book[10010],b[10010];//book标记该数是不是独立幸福数，b标记的是循环过程中衍生出来的数字（用来判断循环是否是死循环）
int n,m,flag;

vector<int>st[10010];//st[i].size()代表i这个数字的衍生出的数字个数，即幸福值

bool prime(int u)//判断该数是不是素数
{
    for(int i=2; i*i<=u; i++)
        if(u%i==0)
            return false;
    return true;
}

void dfs(int u,int k)//u代表当前数字，k代表祖先数字
{
    char a[10010];
    int sum=0;
    sprintf(a,"%d",u);//转化成字符串，方便计算
    for(int i=0; i<strlen(a); i++)
        sum+=(a[i]-'0')*(a[i]-'0');//计算
    if(sum==1)//如果可以达到1，就代表该数是幸福数
        return;
    if(b[sum]==0)//如果这个数字在循环前没出现过
    {
        st[k].push_back(sum);//把衍生出来的数字放进去
        b[sum]=1;//标记这个衍生数字，如果后续再次出现该数字就代表是死循环数字
        book[sum]=1;//标记sum这个数字不是祖先数字，因为它是由k衍生出来的
        dfs(sum,k);//继续递归
    }
    else
    {
        book[k]=1;//如果遇到了死循环数字，就把这个祖先数字标记，代表这个祖先数字不是幸福数字（无法达到1）
        return;
    }
    return;
}

int main()
{
    cin>>n>>m;
    int result=0;
    for(int i=n; i<=m; i++)
    {
        if(book[i]==0)//如果这个数字没被标记过
        {
            memset(b,0,sizeof(b));//初始化衍生数字全为0
            dfs(i,i);//递归
        }
    }
    for(int i=n; i<=m; i++)
        if(book[i]==0)//如果该数字是独立幸福数字
        {
            result=1;
            if(prime(i))//判断该数字是否是素数
                printf("%d %d\n",i,2*(st[i].size()+1));
            else
                printf("%d %d\n",i,st[i].size()+1);
        }
    if(result==0)
        printf("SAD\n");
    return 0;
}