在Python中检查能否从碎片中组成数组的程序
假设我们有一个数组nums,其中所有元素都是唯一的,并且有一个包含不同小数组的另一个数组pieces。我们必须检查是否可以通过按任何顺序串联pieces中的数组来得到主数组nums。但是,我们不允许重新排序每个数组pieces [i]中的元素。
因此,如果输入是nums = [5,1,12,36,2,47,6],pieces = [[2,47,6],[12,36],[1],[5]],则输出将为True,因为我们可以按此顺序将它们串联起来[[5],[1],[12,36],[2,47,6]]来获得主数组。
要解决这个问题,我们将按如下步骤执行:
- temp:一个新列表
-
对于pieces中的每个p,执行以下操作:
- 如果p [0]不在nums中,则返回False
- 否则,返回True
- l:p的大小
-
indx:在nums中索引(p [0])
-
如果从索引indx到indx + l-1的nums子数组与p不同,则返回False
- 否则,将p添加到temp中
- 如果p [0]不在nums中,则返回False
- 如果nums的大小与temp的大小相同,则返回True
- 否则,返回False
更多Python相关文章,请阅读:Python 教程
示例(Python)
让我们看一下以下实现,以便更好地理解。
def solve(nums, pieces):
temp = []
for p in pieces:
if p[0] not in nums:
return False
l = len(p)
indx = nums.index(p[0])
if nums[indx:indx+l] != p:
return False
else:
temp.extend(p)
if len(nums) == len(temp):
return True
else:
return False
nums = [5,1,12,36,2,47,6]
pieces = [[2,47,6],[12,36],[1],[5]]
print(solve(nums, pieces))
输入
[5,1,12,36,2,47,6],[[2,47,6],[12,36],[1],[5]]
输出
True