PAT甲级 1019 普通回文数

原题链接

一个数字如果从前往后读和从后往前读都完全一样，那么这个数就是回文数。

例如，1234321 就是一个回文数。

所有的一位数字都是回文数。

我们一般都是在十进制下考虑回文，但是回文这一概念也可以应用于其他进制下的数字中。

一个整数 N，在 b 进制下，共有 k+1 位，其中第 i 位的数字为 ai,(0≤i≤k)，那么 N=∑ki=0(aibi)。

一般来说，0≤ai<b，并且 ak≠0。

如果对于任意 i，满足 ai=ak−i，那么 N 在 b 进制下就是一个回文数。

0 在任何进制下都表示为 0，且都视为是一个回文数。

现在给定一个整数 N，请你判断它在 b 进制的表示下，是否是一个回文数。

输入格式
共一行包含两个整数 N 和 b。

输出格式
输出共两行。

如果 N 在 b 进制下是回文数，则第一行输出 Yes，否则输出 No。

第二行，输出 N 在 b 进制下的表示，包含 k+1 个整数，要求按顺序输出 ak,ak−1,…,a0，整数之间用空格隔开。

数据范围
0≤N≤109,
2≤b≤109
输入样例1：
27 2
输出样例1：
Yes
1 1 0 1 1
输入样例2：
121 5
输出样例2：
No
4 4 1

我的解法：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e9 + 10;
int n, b;
vector<int> nums;
bool check(){
    for(int i = 0, j = nums.size() - 1; i < j; i ++, j --){
        if(nums[i] != nums[j]){
            return false;
        }
    }
    return true;
}
int main(){
    cin >> n >> b;
    if(!n) nums.push_back(0);
    while(n) nums.push_back(n%b), n/=b;
    if(check()) puts("Yes");
    else puts("No");
    cout << nums.back();
    for(int i = nums.size() - 2; i >= 0; i -- ) cout << " " << nums[i];
    return 0;
}