用于检查我们是否可以在 Python 中从片段形成数组的程序

假设我们有一个数组 nums,其中所有元素都是唯一的,并且有另一个数组,其中包含不同的较小数组,称为pieces。我们必须检查是否可以通过以任何顺序将数组连接成块来获得主数组 nums。但是我们不允许对每个数组pieces[i] 中存在的元素重新排序。

因此,如果输入类似于 nums = [5,1,12,36,2,47,6] 件 = [[2,47,6],[12,36],[1],[5]],那么输出将为 True,因为我们可以按照 [[5], [1], [12,36], [2,47,6]] 的顺序连接它们以获得主数组。

示例(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

猜你喜欢