[PAT乙级]1022. D进制的A+B (20)

1022. D进制的A+B (20)

原题链接
输入两个非负10进制整数A和B(<=230-1)，输出A+B的D (1 < D <= 10)进制数。

输入格式：

输入在一行中依次给出3个整数A、B和D。

输出格式：

输出A+B的D进制数。

输入样例：
123 456 8
输出样例：
1103

注意：

• 这道题我第一次提交发生错误，显示部分正确，我以为是测试数据超出int范围，改成long仍旧是部分正确
• 查看别人的代码后发现，我没有处理sum==0时状况，改正后提交通过

代码：

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

int main()
{
    int A,B,D;
    cin >> A >> B >> D;
    int sum = A+B;
    if(sum == 0){
        cout << 0;
        return 0;
    }
    vector<int> res;
    while(sum != 0){
        int temp = sum%D;
        res.push_back(temp);
        sum /= D;
    }
    for(int i=res.size()-1; i>=0; i--)
        cout << res[i];
    return 0;
}