给定一个包含正数和负数的整数数组,重新排列它,使得所有的负数排在前面,所有的正数排在后面。正数间和负数间的相对顺序保持不变。期望时间复杂度是O(n), 空间复杂度是O(1).
例如: 给定 [-1,2,-2,3,5,-4], 重新排列后变成 [-1,-2,-4,2,3,5]
分析: 1. 最简单的算法是O(n^2), 遇到每个负数都把它移动到数组前面已经排列好的负数部分后面。
2. O(n)的算法暂时没想出来,似乎比较难,下面给出一个O(nlogn)的算法。基本思想类似于Merge Sort. 空间复杂度由于用到递归,是O(logn). 不过可以很容易改写为bottom-up的迭代版本。
void Main() { int[] nums = new int[] {-1, 2, -2, 3, 5, -4}; Reorder(nums, 0, nums.Length-1); } // Define other methods and classes here public void Reorder(int[] nums, int start, int end) { if(start >= end) { return; } int middle = start + (end-start)/2; Reorder(nums, start, middle); Reorder(nums, middle+1, end); Merge(nums, start, middle, end); } private void Merge(int[] nums, int start, int end) { int i = start; while(nums[i] < 0 && i <= end) i++; int j = end; while(nums[j] >= 0 && j >= start) j--; // Shift the negative part to the front if(i < j) { int k = j; while(k>=i && nums[k]<0) k--; Reverse(nums, i, k); Reverse(nums,k+1,j); Reverse(nums,i,j); } } private void Swap(int[] nums, int i, int j) { int t = nums[i]; nums[i] = nums[j]; nums[j] = t; } private void Reverse(int[] nums, int i, int j) { while(i < j) { Swap(nums, i, j); i++; j--; } }
下面是自底向上的迭代版本。空间复杂度是O(1).
public void Reorder(int[] nums) { int n = nums.Length; for(int size = 1; size < n; size += size) { for(int low = 0; low < n-size; low+=size+size) { Merge(nums, low, Math.Min(low+size+size-1, n-1)); } } }
时间: 2024-10-13 15:02:09