最近公司经常有些分享会,前几天有个某婚恋网元老级程序员跳到我们公司并做了次分享,内容是一些算法方面的。
闲来无事动动脑。 偶见一腾讯面试题
一个未排序整数数组,有正负数,重新排列使负数排在正数前面,并且要求不改变原来的 相对顺序 比如: input: 1,7,-5,9,-12,15 ans: -5,-12,1,7,9,15 要求时间复杂度O(N),空间O(1)
这是我的代码,欢迎朋友指正:
public static void txsort(int[] array)
{
int negCount = 0; int length = array.length; int negIndex =0; for(int i=0;i<length;i++) { if(array[i]<0) negCount++; } if(negCount==0) return; for(int i=0;i<length;i++) { int posIndex = (i-negIndex)%(negCount-negIndex) + negIndex; int newIndex = i+negCount - negIndex; if(array[posIndex]<0) { newIndex = negIndex; negIndex++; } int temp = array[posIndex]; array[posIndex] = array[newIndex]; array[newIndex] = temp; if(negIndex==negCount) break; } }
思想如下: 题目要求空间O(1)in-place,时间O(n),且是必须是稳定的。初一看像排序,排序算法中快排的时间复杂度是nlogn。在看一次题目, 只是把正负数分开,不涉及到从大到小。审题很重要。。。。
1 先找出负数的个数N
2 从前到后把元素向后移 负数个位子N, 把后面位置上的数先暂存到前面为负数留的位置上。 当移动 第N+I时,此时N+I 其实是原数组的I, 而N+1 的真实值存在前面的I位置上,所以这里移动的是I 到 N+ I+ N去。
3 遇到负数就往前插 后面的数往后移动 N -1。
4 直到负数全到前面为止
难点: 因为不能开辟新的空间,否则不满足空间复杂度O(1),只能在数组内移动,所以就比较饶。下面把移动演示一下:
0(index) | 1 | 2 | 3 | 4 | 5 |
1 | 7 | -5 | 9 | -12 | 15 |
7 | 1 | 9 | -12 | 15 | |
-5(1 和 -5 交换 暂存-5 到该位置) | |||||
1 | 7 | -12 | 15 | ||
-5 | 9(7和9交换 9暂存到改位置) | ||||
-5(-5原始index 是2, 这里是把原始位置上的-5插入到第0个位置) | 1 | 7 | -12 | 15 | |
9 | |||||
-5 | 1 | 7 | 9 | 15 | |
-12 (把index 3上的9 往后移动 2(负数的总数) - 1(已经存好的负数个数)) 就是把index3 移到index4上 | |||||
-5 | -12 | 1 | 7 | 9 | 15 |
要注意int posIndex = (i-negIndex)%(negCount-negIndex) + negIndex; 中的i-negIndex. 去掉已移动。