[已解决]计算嵌套列表某一层次的元素数量

ylhhh

【问题描述】

输入一个嵌套列表，再输入层数，求该层的元素个数。如果求第二层次的元素个数，但是第二层次还有第三层次嵌套，则嵌套列表算一个元素。

【输入形式】

第一行输入嵌套列表，包含方括号，逗号分隔，每个元素是整数

第二行输入需要计算的层次，整数，大于0
【输出形式】

整数，计算所得的个数
【样例输入】

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

2

【样例输出】

4
【样例说明】

在第二层次 有3,4，[5,6], 7 等4个元素

最佳答案

月排行榜 / 总排行榜

isdkz

2022-3-6 16:17:42

1. def func(arr, level = 1):
   
2.         res = 0
   
3.         if level == 1:
   
4.             res += len(arr)
   
5.         else:
   
6.             for each in arr:
   
7.                 if type(each) == list:
   
8.                     res += func(each, level - 1)
   
9.         return res
   
10. 
    
11. arr = [1,2,[3,4,[5,6],7],8]
    
12. 
    
13. print(func(arr, 2))

复制代码

跳转到最佳答案楼层

傻眼貓咪

1. arr = [1, 2, [3, 4, [5, 6], 7], 8]
   
2. 
   
3. def func(arr, res, level = 1):
   
4.     if res != level:
   
5.         return func(arr, res, level + 1)
   
6.     else:
   
7.         return len(arr)
   
8. 
   
9. print(func(arr, 2))

复制代码

isdkz

1. def func(arr, level = 1):
   
2.         res = 0
   
3.         if level == 1:
   
4.             res += len(arr)
   
5.         else:
   
6.             for each in arr:
   
7.                 if type(each) == list:
   
8.                     res += func(each, level - 1)
   
9.         return res
   
10. 
    
11. arr = [1,2,[3,4,[5,6],7],8]
    
12. 
    
13. print(func(arr, 2))

复制代码

ylhhh

傻眼貓咪 发表于 2022-3-6 16:02

大佬，我没有很看得懂

ylhhh

1. def func(nums,n):
   
2.     if n == 1:
   
3.         return len(nums)
   
4.     if n > 1:
   
5.         for i in nums:
   
6.             if type(i) is list:
   
7.                 n = n - 1
   
8.                 return func(i,n)
   
9. 
   
10. nums = eval(input())
    
11. n = int(input())
    
12. print(func(nums,n))

复制代码

ylhhh

ylhhh 发表于 2022-3-6 16:32

这是我自己写的一个程序

isdkz

ylhhh 发表于 2022-3-6 16:29
大佬，我没有很看得懂

你可以用多个不同的列表来测试一下

傻眼貓咪

ylhhh 发表于 2022-3-6 16:29
大佬，我没有很看得懂

抱歉，刚才写太快了，有错，已修改：

1. arr = [1, 2, [3, 4, [5, 6], 7], 8]
   
2. 
   
3. def func(arr, res, level = 1):
   
4.     if res != level:
   
5.         for each in arr:
   
6.             if type(each) != int:
   
7.                 return func(each, res, level + 1)
   
8.     else:
   
9.         return len(arr)
   
10. 
    
11. print(func(arr, 2))

复制代码

ylhhh

傻眼貓咪 发表于 2022-3-6 16:34
抱歉，刚才写太快了，有错，已修改：

谢谢大佬

isdkz

傻眼貓咪 发表于 2022-3-6 16:34
抱歉，刚才写太快了，有错，已修改：

你这个还是有点问题：比如你用 [1, [2], [3, 4, [5, 6], 7, 5, 8], [8]] 这个列表来测试得到的是结果 1，

你的只得出了第一个嵌套列表的长度

isdkz

ylhhh 发表于 2022-3-6 16:32

你的代码跟 8 楼的是一个问题， 像 [1, [2], [3, 4], [8]] 这样的列表里面有三个嵌套列表，

你没有统计所有第二层列表的，就只得到了第一个 二层列表 的长度。

傻眼貓咪

isdkz 发表于 2022-3-6 16:38
你这个还是有点问题：比如你用 [1, [2], [3, 4, [5, 6], 7, 5, 8], [8]] 这个列表来测试得到的是结果 1， ...

我也是有想过这点，但是如果真的是这样，就不符合逻辑了，比如：
[1, [2], [3, 4, [5, 6], 7, 5, 8], [8]]

测试第 2 层，那么需要测试 [2] 还是  [3, 4, [5, 6], 7, 5, 8] 还是 [8] 呢？
该不会全部加起来吧？

isdkz

傻眼貓咪 发表于 2022-3-6 16:46
我也是有想过这点，但是如果真的是这样，就不符合逻辑了，比如：
[1, [2], [3, 4, [5, 6], 7, 5, 8], [ ...

那几个不是都属于第二层列表吗？

傻眼貓咪

isdkz 发表于 2022-3-6 16:48
那几个不是都属于第二层列表吗？

也是，可能我的思路有错吧，哈哈，我是把它当成每层只有一个列表元素