前面学习了完美洗牌问题
又写了一个证明
进一步思考了其他的一些问题:
完美洗牌问题: 给定的输入a1, a2, a3, ……aN, b1,b2,……bN,输出b1,a1,b2,a2,b3,a3…… bN,aN
(1) 如果要求输出是a1,b1,a2,b2……aN,bN怎么办?
这个问题在学习的时候已经考虑过,只是觉得如果先把a部分和b部分交换掉,或者最后再交换相邻的一组两个位置的方法不够美观。
现在想想可以这样,原数组第一个和最后一个不变,中间的2 * (n - 1)项用原始的标准完美洗牌算法做就可以了。
(2) 完美洗牌问题的逆问题:
给定b1,a1,b2,a2,……bN,aN, 输出a1,a2,a3,……aN,b1,b2,b3,……bN
这相当于把偶数位上的数放到一起,奇数位上的数放到一起。
关键问题: 我们需要把cycle_leader算法改一下,沿着圈换回去。改造后的叫reverse_cycle_leader,代码如下:
//逆变换,数组下标从1开始,from是圈的头部,mod是要取模的数 mod 应该为 2 * n + 1,时间复杂度O(圈长)
void reverse_cycle_leader(int *a,int from, int mod) {
int last = a[from],next, i;
for (i = from;;i = next) {
next = i * 2 % mod;
if (next == from) {
a[i] = last;
break;
}
a[i] = a[next];
}
}
按照完美洗牌算法,我们同样把数分为m和(n - m)两部分。
假设我们把前面若干项已经置换成先a后b的形式了,现在把这m项也置换成先a后b的形式,我们需要把这m项中的a部分换到前面去,这里需要一个循环右移,还要知道以前处理了多长。总之,这个逆shuffle算法需要小心实现一下,代码如下:
//逆shuffle 时间O(n),空间O(1)
void reverse_perfect_shuffle3(int *a,int n) {
int n2, m, i, k, t, done = 0;
for (;n > 1;) {
// step 1
n2 = n * 2;
for (k = 0, m = 1; n2 / m >= 3; ++k, m *= 3)
;
m /= 2;
// 2m = 3^k - 1 , 3^k <= 2n < 3^(k + 1)
for (i = 0, t = 1; i < k; ++i, t *= 3) {