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

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

输入格式:

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

输出格式:

在一行中输出元素个数

输入样例:

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

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

输出样例:

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

2

题解：

  1. s=input()
  2. input_layer=eval(input())
  3. cur_layer,ele_count,flag=0,0,1
  4. for i in range(len(s)):
  5.     if s[i]=='"' or s[i]=="'":
  6.         flag=-flag
  7.     if flag==1:
  8.         if s[i]=='[':
  9.             cur_layer+=1
  10.         elif s[i]==']':
  11.             cur_layer-=1
  12.         if (input_layer==cur_layer) and (s[i].isdigit()) and \
  13.         not s[i+1].isdigit() and s[i+1]!='.':
  14.             ele_count+=1
  15. print(ele_count)

 

注：本文转载自blog.csdn.net的amateur_python的文章"https://blog.csdn.net/amateur_python/article/details/119135253"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！