HDU1013 Digital Roots

这题本来想着用模拟来着，于是写着写着发现测试用例子没问题但提交后会WA，那就说明了输入一定是大整数了，如果使用大整数继续模拟那么应该会AC的。但我想看看是否有更加精妙的解法，于是发现大神的博客大致是这么写道：通过找规律得到“roots = (n - 1) % 9 + 1”，规律就不再推到了。贴上我之前错误的代码：

#include <iostream>
using namespace std;

// 搜索一个int的Digital root
long search(long n) {
    long sum = 0;
    if (n < 10) return n;
    else {
        while (n != 0) {
            sum += n % 10;
            n /= 10;
        }
        search(sum);
    }
}
int main()
{
    long n;
    while (cin >> n && n) {
        cout << search(n) << endl;
    }
    return 0;
}

以及通过数论的代码：

#include <iostream>
#include <cstring>
using namespace std;

int main()
{
    int sum;
    char input[1000 + 5];
    cin >> input;
    while (input[0] != '0') {
        sum = 0;
        for (int i = 0; i < strlen(input); ++i){
            sum += input[i] - '0';
        }
        // 接下来继续模拟也可以，因为sum的规模相当小，但在此使用公式
        cout << (sum - 1) % 9 + 1 << endl;
        cin >> input;
    }
    return 0;
}