L2-029 特立独行的幸福 (25 分) find()、vector、map

题目

L2-029 特立独行的幸福 (25 分)
对一个十进制数的各位数字做一次平方和，称作一次迭代。如果一个十进制数能通过若干次迭代得到 1，就称该数为幸福数。如 19 经过 1 次迭代得到 82，2 次迭代得 68，3 次迭代得 100，最后到 1。则 19 就是幸福数。一个幸福数迭代到 1 的过程中经过的数字都是幸福数，它们的幸福是依附于初始数字的。例如 82、68、100 的幸福是依附于 19 的。而一个特立独行的幸福数，是在一个有限的区间内不依附于任何其它数字的；其独立性就是依附于它的的幸福数的个数。如果这数是素数，独立性加倍。如 19 就是一个特立独行的幸福数，其独立性为2×4=8。
另一方面，如果一个大于1的数字经过数次迭代后进入了死循环，那这个数就不幸福。

输入格式：
输入在第一行给出闭区间的两个端点：1<A<B≤10^​4
输出格式：
按递增顺序列出给定闭区间 [A,B] 内的所有特立独行的幸福数和它的独立性。没有幸福数，输出 SAD。

输入样例 1：

10 40

输出样例 1：

19 8
23 6
28 3
31 4
32 3

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

find（）

find函数用于查找数组中的某一个指定元素的位置，在algorithm里
find （数组名 + 起始查找元素的位置， 数组名 + 结束查找的元素位置， 想要查找的元素）
如：有一个数组[0, 0, 5, 4, 4]；
问：元素5的在什么位置，find函数 返回值 为 2；

分析

如果这个数还是个素数，则其独立性加倍，就是2的意思。19 独立性为4×2=8是因为要经过4个数才会变成1,19还是个素数就2

代码

#include<iostream>
#include<vector>
#include<math.h>
#include<map>
#include<algorithm>
using namespace std;

int isPrime(int x) {
	for (int i = 2; i <= sqrt(x); i++)
		if (x%i == 0)
			return 1;
	return 2;//是素数的话*2
}

int main() {
	int l, r, t, sum, flag = 0, appear[10001] = { 0 };
	map<int, int> link;
	cin >> l>> r;
	for (int i = l; i <= r; i++) {
		t = i;
		vector<int> road;
		while (t != 1) {
			sum = 0;
			do	sum += pow(t % 10, 2);	while (t /= 10);	//各位数求平方和
			t = sum;
			if (find(road.begin(), road.end(), sum) != road.end())  
				break;
			road.push_back(sum);
			appear[t] = 1;   //标记出现过的数
		}
		if (t == 1) 
            link[i] = road.size(); //存储辛福数
	}
	/*for (auto &it : link)
        if (!appear[it.first] && ++flag)  //排除非独立的幸福数
			cout << it.first << " " << isPrime(it.first)*it.second << endl;*/
    for(map<int,int>::iterator it=link.begin();it!=link.end();it++)
		if (!appear[it->first] && ++flag)  //排除非独立的幸福数 it->first 是独立数，it->second是要经过的数的个数
			cout << it->first << " " << isPrime(it->first)*it->second << endl;//不能写成.
	if (!flag) cout << "SAD" << endl;
	return 0;
}