PAT1 1019 General Palindromic Number

题目链接
我的github

题目大意

给一个10进制数$N$和$D$，求$N$转成$D$进制后是否是回文数（如果一个数从前往后和从后往前看都是一样的，就称这个数是回文数）只有一位的数也是回文数

输入

每组包含一个测试样例，每个样例第一行就是两个正整数$N$($0\le N\le 10^9$)，$D$($2\le D\le 10^9$)

输出

如果这个数转成$D$进制后是回文数就先输出Yes否则输出No，之后一行输出$N$在$D$进制下的每一位，每一位之间用空格分开，行尾没空格

输入样例

27 2

121 5

输出样例

Yes
1 1 0 1 1

No
4 4 1

解析

转完进制后判断一下是不是回文数就行

def tran(n, d):
    res = list()
    while n:
        res.append(n % d)
        n //= d
    return res[::-1]


def solve():
    n, d = map(int, input().split())
    res = tran(n, d)
    flag = True
    for i in range(len(res)):
        if len(res) - i - 1 >= i:
            if res[i] != res[len(res) - 1 - i]:
                flag = False
                break
    print("Yes" if flag else "No")
    for i in range(len(res)):
        print(res[i], end=("\n" if i == len(res) - 1 else " "))


if __name__ == "__main__":
    solve()