剑指offer

面试题21. 调整数组顺序使奇数位于偶数前面

2020-03-13  本文已影响0人  人一己千

题目

输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有奇数位于数组的前半部分,所有偶数位于数组的后半部分。

示例:

输入:nums = [1,2,3,4]
输出:[1,3,2,4] 

注:[3,1,2,4] 也是正确的答案之一。

提示:

1 <= nums.length <= 50000
1 <= nums[i] <= 10000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diao-zheng-shu-zu-shun-xu-shi-qi-shu-wei-yu-ou-shu-qian-mian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法

最简单的解法:

class Solution:
    def exchange(self, nums: List[int]) -> List[int]:
        i = 0
        j = len(nums) - 1
        while i < j:
            while i < j and nums[i] & 1 == 1:
                i += 1
            while i<j and  nums[j] & 1 == 0:
                j -= 1
            nums[j],nums[i] = nums[i],nums[j]
        return nums

值得注意的是在i++ 的过程中不能忘了条件判断以免越界。

剑指offer中提到了应该要把判断条件提取出了做一个函数以增加扩展性,本来我一开始的写法想用一个函数来判断,但是又想到单独把这么简短的一个条件拿出来写函数不值当。如果真有这个需求的话再写吧,不然照着为了提前考虑扩展性的思路,判断越界也单独做一个函数?
还是应该Pythonic一点。

总结

主要是不要越界了。

上一篇下一篇

猜你喜欢

热点阅读