http://www.jiuzhang.com/problem/39/
给一个数组A,和一个整数k,将数组分成两个部分(你可以移动数组中的数),使得左边部分的数都<k,右边部分的数都>=k。要求使用O(1)的额外空间,和O(n)的时间。
利用快速排序的思想。用两根指针i和j,一根指向头,一根指向尾,将头移动到第一个不满足A[i]<k的数,将尾巴移动到第一个不满足A[j]>=k的数,交换两个数,然后重复往下查找不满足A[i]<k和A[j]>=k的数,继续交换。直到两根指针碰头。python代码如下:
i, j = 0, len(A) -1
while i <= j:
while i < j and A[i] < k:
i += 1
while i < j and A[j] >= k:
j -= 1
if i <= j:
A[i], A[j] = A[j], A[i] #swap
i += 1
j -= 1
本题的考点是快排。如果写过快排或者明白快排的原理,很快就会明白这个题目只是快速排序中间的一个部分。两根指针一头一尾的这种方式,往期的题目中,第32题小球排序也是利用了类似的思路。读者可以将两个题目联系起来,找到共性,有利于思路的整理和归纳。
原文地址:http://blog.csdn.net/jiuzhang_ninechapter/article/details/45669557