最近公司经常有些分享会,前几天有个某婚恋网元老级程序员跳到我们公司并做了次分享,内容是一些算法方面的。
闲来无事动动脑。 偶见一腾讯面试题
一个未排序整数数组,有正负数,重新排列使负数排在正数前面,并且要求不改变原来的 相对顺序 比如: 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. 去掉已移动。