[C++] 백준 14954번 Happy Number

각 자리의 수를 제곱해서 더하는 것을 반복했을 때
최종적으로 1이 나오는 수는 HAPPY한 수입니다.
1이 나오지 않고 반복되는 수는 UNHAPPY한 수입니다.

나오는 수들을 vector에 넣어서 저장해두고
DFS를 통해서 1이 나올때 까지 계산을 반복합니다.
만약 DFS도중 vector에 있는 수가 나올 경우 UNHAPPY한 상태이며,
1이 나오는 경우면 HAPPY한 상태라고 판단할 수 있습니다.

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
using namespace std;

int n;
vector<int> check;

int solve(int x)
{
	check.push_back(x);
	int result = 0;

	while (x != 0) {
		result += pow(x % 10, 2);
		x /= 10;
	}

	if (result == 1) {
		return 1;
	}

	if (find(check.begin(), check.end(), result) != check.end()) {
		return 0;
	}

	return solve(result);
}

int main(int argc, char* argv[])
{
	cin >> n;

	solve(n) == 1 ? cout << "HAPPY" << endl : cout << "UNHAPPY" << endl;
	
	return 0;
}