[백준] 6378번 디지털 루트

https://www.acmicpc.net/problem/6378

6378번: 디지털 루트

처음에는 이렇게 제출했었는데 틀렸다.

#include <iostream>
#include <vector>
#include <string>
using namespace std;


int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	while (true) {
		int n;
		cin >> n;
		if (n == 0) break;
		while (true) {
			vector<int> v;
			while (n != 0) {
				v.push_back(n % 10);
				n /= 10;
			}
			int total = 0;
			for (int i : v) {
				total += i;
			}
			if (total < 10) {
				cout << total << '\n';
				break;
			}
			else {
				n = total;
			}
		}
	}
	return 0;
}

보니까 숫자는 최대 1000자리수라고 했다.

그러니 당연히 아무리 범위가 큰 타입을 써도 오버플로우가 날 수밖에 없었다.

따라서 문자열로 받아서 풀었다.

#include <iostream>
#include <vector>
#include <string>
#include <map>
using namespace std;


int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int numbers[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
	char charNumbers[10] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' };
	string stringNumbers[10] = { "0", "1", "2", "3", "4", "5", "6", "7", "8", "9" };
	map<char, int> m;
	map<int, string> sm;
	for (int i = 0; i < 10; i++) {
		m[charNumbers[i]] = numbers[i];
		sm[numbers[i]] = stringNumbers[i];
	}
	while (true) {
		string n;
		cin >> n;
		if (n == "0") break;
		while (true) {
			int total = 0;
			for (int i = 0; i < n.size(); i++) {
				total += m[n[i]];
			}
			if (total < 10) {
				cout << total << '\n';
				break;
			}
			else {
				n = "";
				while (total != 0) {
					n.append(sm[total % 10]);
					total /= 10;
				}
			}
		}
	}
	return 0;
}