在Python中查找元素数量,A中的元素严格小于B中至少k个元素
假设我们有两个数字列表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
- 如果i < B [k-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