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

    • l := p的大小

    • indx := p[0]在nums中的索引

    • 如果nums从索引indx到indx+l-1的子数组与p不同，则

      • 返回False

    • 否则

      • 将p添加到temp之后

  • 如果nums的大小与temp的大小相同，则

    • 返回True

  • 否则

    • 返回False

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

更新于：2021年5月17日

116 次浏览

开启你的职业生涯

完成课程获得认证

开始学习
广告