PTA L4-112 不变初心数(Python)

不变初心数是指这样一种特别的数，它分别乘 2、3、4、5、6、7、8、9 时，所得乘积各位数之和却不变。例如 18 就是这样的数：18 的 2 倍是 36，3+6=9；18 的 3 倍是 54，5+4=9；…… 18 的 9 倍是 162，1+6+2=9。对于 18 而言，9 就是它的初心。本题要求你判断任一个给定的数是否有不变的初心。

输入格式：

输入在第一行中给出一个正整数 N（≤ 100）。随后 N 行，每行给出一个不超过 105 的正整数。

输出格式：

对每个给定的数字，如果它有不变的初心，就在一行中输出它的初心；否则输出 NO。

输入样例：

4
18
256
99792
88672

输出样例：

9
NO
36
NO

代码如下：

n=int(input())
s = 0
a = []
for i in range(n):
    m=int(input())
    for j in range(2,10):
        temp = m * j
        for k in str(temp):
            s=s+int(k)
        a.append(s)
        s=0
    a=set(a)
    a=list(a)
    if len(a)==1:
        print(a[0])
    else:
        print('NO')
    a=[]