0
点赞
收藏
分享

微信扫一扫

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

奔跑的酆 2022-01-27 阅读 48
python

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

输入格式:
第一行输入列表 第二行输入层数

输出格式:
在一行中输出元素个数

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

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

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

2

题目集全集传送门

a = input()
n = int(input())
b = ''.join(a)
count = 0
cnt = 0
for i in range(len(b)):
    if b[i] == '[':
        count += 1
    elif b[i] == ']':
        count -= 1
    if n == count and b[i] != '[' and b[i] != ']' and b[i] != ',' and b[i + 1].isdigit() == False:
        cnt += 1
print(cnt)
举报

相关推荐

0 条评论