zoukankan      html  css  js  c++  java
  • 第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)
     
  • 相关阅读:
    牛人对IC验证的独特理解(转载)
    soc验证扫盲(转载)
    .vimrc
    matchit匹配
    格式化verilog/systemverilog代码插件
    gvim plugin管理
    .alias
    .cshrc
    get_result --perl
    run_regress --perl
  • 原文地址:https://www.cnblogs.com/dreamcoding/p/12694572.html
Copyright © 2011-2022 走看看