用 Python 编写程序,查找 A 中元素的数量严格小于 B 中至少 k 个元素的数量

pythonserver side programmingprogramming

假设我们有两个数字列表 A 和 B,以及另一个值 k,我们必须找到 A 中元素的数量严格小于 B 中至少 k 个元素的数量。

因此,如果输入为 A = [6, -2, 100, 11] B = [33, 6, 30, 8, 14] k = 3,则输出将为 3,因为 -2、6 和 11 严格小于 B 中的 3 个元素。

为了解决这个问题,我们将遵循以下步骤 −

  • 如果 k 与 0 相同,则
    • 返回 A 的大小
  • 按相反顺序对 B 进行排序
  • ct := 0
  • 对于 A 中的每个 i,执行
    • 如果 i < B[k - 1],则
      • ct := ct + 1
  • 返回 ct

让我们看看下面的实现以便更好地理解 −

示例

class Solution:
   def solve(self, A, B, k):
      if k == 0:
         return len(A)
      B.sort(reverse=True)
      ct = 0
      for i in A:
         if i < B[k - 1]:
            ct += 1
      return ct
ob = Solution()
A = [6, -2, 100, 11]
B = [33, 6, 30, 8, 14]
k = 3 print(ob.solve(A, B, k))

输入

[6, -2, 100, 11], [33, 6, 30, 8, 14], 3

输出

3

相关文章


有用资源