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;
}