题目
输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有奇数位于数组的前半部分,所有偶数位于数组的后半部分。
示例:
输入: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一点。
总结
主要是不要越界了。
网友评论