日常练习：Lintcode31. 数组划分 Python

自从上次周六做了那些题之后，就累了，断更三天了。
今天补一道题吧。

题目：

给出一个整数数组 nums 和一个整数 k。划分数组（即移动数组 nums 中的元素），使得：
所有小于k的元素移到左边
所有大于等于k的元素移到右边
返回数组划分的位置，即数组中第一个位置 i，满足 nums[i] 大于等于 k。

注意事项:

你应该真正的划分数组 nums，而不仅仅只是计算比 k 小的整数数，如果数组 nums 中的所有元素都比 k 小，则返回 nums.length。

样例:

给出数组 nums = [3,2,2,1] 和 k = 2，返回 1.

上面那个注意事项，没做到，就先不硬排序了。

代码

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
        if(not nums):
            return 0
        nums.sort()
        if(max(nums)<k):
            return len(nums)
        else:
            for i in range(len(nums)):
                if(nums[i] >= k):
                    return i

没啥好说的…挑的简单题，不是很难，也没按要求来，就这样吧