1. 问题描述
Given an array nums of integers and an int k, partition the array (i.e move the elements in “nums”) such that:
- All elements < k are moved to the left
- All elements >= k are moved to the right
Return the partitioning index, i.e the first index i nums[i] >= k.
Challenge
Can you partition the array in-place and in O(n)?
2. my solution
2.1 我的思路
首先, 我之后在找别的solution的时候才发现大家没有注意到下面几个点
- 题目中并没有要求返回数组, 也就是数组并不需要更改
- 题目中没有要求数组一定是有序的, 只需要小的在左边, 大的在右边即可
少了上面两个误区, 可以构造以下思路
建立一个新的数组大小与输入数组一样, 从0开始遍历数组, 如果比k小就放到最左边, 如果比k大就放到最右边
- 时间复杂度: O(n) (完成挑战)
2.2 代码实现
public class Solution {
/**
* @param nums: The integer array you should partition
* @param k: An integer
* @return: The index after partition
*/
public int partitionArray(int[] nums, int k) {
// write your code here
int lo = 0 ;
int hi = nums.length- 1;
int[] r = new int[nums.length];
for(int i = 0 ;i<nums.length; i++)
{
if(nums[i] < k)
r[lo++] = nums[i];
else
r[hi--] = nums[i];
}
return lo;
}
}
2.3 运行结果
可以看到结果还是不错的