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

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

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

from collections import deque


class Solution:
    """
    @param nums: The integer array you should partition
    @param k: An integer
    @return: The index after partition
    """

    def partitionArray(self, nums, k):
        # write your code here
        new_nums_list = deque(nums)
        result_list = deque()
        index = 0
        for num in new_nums_list:
            if num < k:
                result_list.appendleft(num)
            else:
                result_list.append(num)
        for idx in result_list:
            if idx >= k:
                index = result_list.index(idx)
                break
            else:
                index = len(result_list)