L2-029 特立独行的幸福 (25分) c++代码

这题想了很久，自己慢慢搭建思路然后一步步实现，完整敲出来后一遍过的感觉超级棒！！！！独立思考感觉真好~~

对一个十进制数的各位数字做一次平方和，称作一次迭代。如果一个十进制数能通过若干次迭代得到 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<iostream>
//#include<cstring>
//#include<cmath>
//#include<algorithm>
//#include<ctype.h>
//#include<stdio.h>
//#include<map>
#include <bits/stdc++.h>
//typedef long long ll;
//const ll N=5e5+5;

using namespace std;

vector<vector<int>> ans;
vector<int> v;
int f[10010];
int ff[10010];

int isPrime(int num){//是素数 独立性*2
	if(num<2) return 1;
	for(int i=2;i<=sqrt(num);i++){
		if(num%i==0) return 1;
	}
	return 2;
}

int dfs(int num){
	if(num==1) return 1;
	f[num]=1;
	v.push_back(num);
	if(ff[num]==1) return 0;//出现循环数字
	ff[num]=1;
	int n=0;
	while(num){
		n+=pow((num%10),2);
		num/=10;
	}
	return dfs(n);
}

bool check(){
	for(int i=0;i<(int)ans.size();i++){
		if(ff[ans[i][0]]==1){//ans[i]是v的子集，替换
			ans[i]=v;
			return true;
		}
	}
	return false;
}

bool cmp(vector<int> a,vector<int> b){
	return a[0]<b[0];
}

int main()
{
	int a,b;
	cin>>a>>b;
	for(int i=a;i<=b;i++){
		v.clear();
		if(f[i]==0){//数字未出现过
			memset(ff,0,sizeof ff);
			if(dfs(i)){//是幸福数字
				if(!check()) ans.push_back(v);
			}
		}
	}
	int f=0;
	sort(ans.begin(),ans.end(),cmp);
	for(int i=0;i<(int)ans.size();i++){
		cout<<ans[i][0]<<" "<<isPrime(ans[i][0])*(int)ans[i].size()<<endl;
		f=1;
	}
	if(f==0) cout<<"SAD";
    return 0;
}