代码改变世界

Modified 2 color sort

2010-12-05 01:33 by Sun Yongyue, ... 阅读, ... 评论, 收藏, 编辑

不知道原来的问题是怎么样子,修改后大概题意就是说有一个由0和1组成的数据要做到先在奇数位放置0,偶数位放置1,多出来的部分再放到最后边。有一个限制条件是原地地交换,也就是原则上不能使用额外空间(当然O(1)的变量还是可以用的)。

比如说输入{0,1,1,0,1,1,0}就变换成{0,1,0,1,0,1,1}

 

好的,没有现成算法,这种问题我喜欢。其实还是可以从其他算法中借鉴一些思想,还记得Knuth给的一个快排吗?

 

eidx = 0; // even index

oidx = 1; // odd index

while (eidx < len && oidx < len)

{

    while (eidx < len && array[eidx] == 1) eidx += 2;

    while (oidx < len && array[oidx] == 0) oidx += 2;

 

    if (eidx < len && oidx < len) swap(eidx, oidx);

}

 

if (eidx < len)

{

    p = eidx;

    q = len & 0xFFFFFFFE;

    while (p < q)

    {

        while (p < q && array[p] == 1) p += 2;

        while (p < q && array[q] == 0) q -= 2;

        if (p < q) swap(p, q);

    }

}

else if (oidx < len)

{

    // similar to above

}

时间复杂度O(n),空间复杂度O(1)