python 题目 给出一个整数数组 nums 和一个整数 k

我会带着你远行 2022-05-22 10:04 289阅读 0赞

给出一个整数数组 nums 和一个整数 k。划分数组（即移动数组 nums 中的元素），使得：

所有小于k的元素移到左边
所有大于等于k的元素移到右边
返回数组划分的位置，即数组中第一个位置 i，满足 nums[i] 大于等于 k。

  1. from collections import deque
  4. class Solution:
  5.     """
  6.     @param nums: The integer array you should partition
  7.     @param k: An integer
  8.     @return: The index after partition
  9.     """
  11.     def partitionArray(self, nums, k):
  12.         # write your code here
  13.         new_nums_list = deque(nums)
  14.         result_list = deque()
  15.         index = 0
  16.         for num in new_nums_list:
  17.             if num < k:
  18.                 result_list.appendleft(num)
  19.             else:
  20.                 result_list.append(num)
  21.         for idx in result_list:
  22.             if idx >= k:
  23.                 index = result_list.index(idx)
  24.                 break
  25.             else:
  26.                 index = len(result_list)

发表评论

表情：
评论列表 （有 0 条评论，289人围观）

还没有评论，来说两句吧...

相关阅读