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)
     
  • 相关阅读:
    Python 在线基础课程
    单链表实现两个集合的交并补
    矩阵奇异值分解SVD
    动态规划0-1背包问题
    Java数字反转
    简单SqL语句
    HTML和CSS
    while和for循环
    C#读取和写入配置文件
    C# 添加系统计划任务方案
  • 原文地址:https://www.cnblogs.com/dreamcoding/p/12694572.html
Copyright © 2011-2022 走看看