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

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

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

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

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

[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)