第6章-6.求指定层的元素个数 (40分)

输入一个嵌套列表,再输入层数,求该层的数字元素个数。

输入格式:

第一行输入列表 第二行输入层数

输出格式:

在一行中输出元素个数

输入样例:

在这里给出一组输入。例如:

[1,2,[3,4,[5,6],7],8]
3
 

输出样例:

在这里给出相应的输出。例如:

2
 1 # 求指定层的元素个数
 2 # Author: cnRick
 3 # Time  : 2020-4-13
 4 def getSum(items,depth,n):
 5     if type(items) == int:
 6         if type(items) == int:
 7             if depth == n:
 8                 return 1
 9             else:
10                 return 0
11     else:
12         if (type(items) == tuple) or (type(items) == list):
13             result = 0
14             for i in range(len(items)):
15                 result += getSum(items[i],depth+1,n)
16             return result
17         else:
18             return 0
19 
20 items = eval(input())
21 n = int(input())
22 result = 0
23 for i in range(len(items)):
24     result = result + getSum(items[i],1,n)
25 print(result)

 

 
posted @ 2020-04-13 22:32  cnRicky  阅读(1100)  评论(0编辑  收藏  举报