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

解决一个整数数组划分问题,目标是将小于k的元素移到左边,大于等于k的元素移到右边,并返回第一个大于等于k的元素的位置。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

题目:

给出一个整数数组 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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值