O(n)
:( 还是不会数位 dp
只需要统计各个数位和对应的数量,男女由于数量相同,直接平方再求和即可
from math import sqrt,ceil,gcd,log;re=lambda:map(int,input().strip().split()) n, = re() cnt = [0] * 46 for i in range(1, n + 1): a = sum(int(j) for j in str(i)) cnt[a] += 1 print(sum(i * i for i in cnt))
注册一个 南阳理工学院OJ 通用账户,您就可以在我们提供的所有在线评测服务上提交代码、参与讨论。
使用您的 南阳理工学院OJ 通用账户