PAT(B) 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

注意不要忘记考虑A B之和为0的情况

#include <iostream>
using namespace std;
int main()
{
    int num1,num2,D,sum;
    cin>>num1>>num2>>D;

        sum = num1 + num2;
        int i = 0;
        int a[40];
        while (sum)
        {
            a[i] = sum%D;
            sum = sum / D;
            i = i + 1;
        }
        if(num1+num2==0)cout<<"0";//为零时也要输出
        for(int j=0;j<i;j++)
        cout<<a[i-j-1];//逆序输出
    return 0;
}