Python-nowcoder数字之和

题目描述

对于给定的正整数 n，计算其十进制形式下所有位置数字之和，并计算其平方的各位数字之和。
输入描述:
每行输入数据包括一个正整数n(0<n<40000)
输出描述:
对于每个输入数据，计算其各位数字之和，以及其平方值的数字之和，输出在一行中，之间用一个空格分隔，但行末不要有空格。

解题思路

说实话我也看题目看了半天，举例如下：若正整数为12，十进制下所有位置的数字为1，2，之和即为1+2=3；其平方即12的平方为144，之和为1+4+4=9。

1、重点在拆解数字上，需要把正整数及其平方数进行分解，原有笨方法：

a = n % 10
b = n // 10 % 10
c = n // 100 % 10
d = n // 1000 % 10
e = n // 10000 % 10
num1 = a+b+c+d+e
# print(a, b, c, d, e)
print(num1)

但是如上方法在用于10位数时则太过冗余，故使用循环解决

2、看到10、100、1000应该能想到 10 的阶乘，利用i做循环：

def sumNum(n):
	num1 = num2 = 0
	for i in range(10):
		if i == 0:
			temp = n % 10
		else:
			temp = n // (10**i) % 10
		# print(temp)
		num1 += temp
	# print(num1)

	m = n*n
	for i in range(10):
		if i == 0:
			temp = m % 10
		else:
			temp = m // (10**i) % 10
		# print(temp)
		num2 += temp
	# print(num2)
	return str(num1)+" "+str(num2)

num = int(input())
print(sumNum(num))