7-5 列表元素個數的加權和(1) (40 分)
輸入一個嵌套列表,嵌套層次不限,根據層次,求列表元素的加權個數和。第一層每個元素算一個元素,第二層每個元素算2個元素,第三層每個元素算3個元素,第四層每個元素算4個元素,...,以此類推!
輸入格式:
在一行中輸入一個列表。
輸出格式:
在一行中輸出加權元素個數值。
輸入樣例:
在這里給出一組輸入。例如:
[1,2,[3,4,[5,6],7],8]
輸出樣例:
在這里給出相應的輸出。例如:
15
a = eval(input())
def fn(base, power):
if type(base) == int:
return power
else:
return sum(fn(i, power+1) for i in base)
print(fn(a, 0))