九章算法面试题39 分割数组

九章算法官网-原文网址

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题小球排序也是利用了类似的思路。读者可以将两个题目联系起来，找到共性，有利于思路的整理和归纳。