完美洗牌问题(1)

完美洗牌问题(1)

题目描述

给定一个长度为偶数的数组arr，长度记为$2\ast N$。前n个为左部分，后n个为右部分。arr可以表示为 { L 1 , L 2 . . . L n − 1 , L n , R 1 , R 2 . . . R n − 1 , R n } \{L_1,L_2...L_{n-1},L_n,R_1,R_2...R_{n-1},R_n \} {L1​,L2​...Ln−1​,Ln​,R1​,R2​...Rn−1​,Rn​},请将数组调整成 { R 1 , L 1 , R 2 , L 2 , . . . R n , L n } \{ R_1,L_1,R_2,L_2,...R_n,L_n \} {R1​,L1​,R2​,L2​,...Rn​,Ln​}。

输入描述：

输入有包含两行，第一行包含一个整数n（$1\le n\le 10^5$ ，保证n为偶数），代表arr长度。第二行包含n个整数，代表数组$arr(1\le ar{r}_i\le 10^8)$。

输出描述：

输出一行n个整数，代表调整后的数组arr。

示例1

输入

6
1 2 3 4 5 6

输出

4 1 5 2 6 3

备注：

时间复杂度$O(N)$，额外空间复杂度$O(1)$。

---

题解：

额外空间复杂度决定着我们只能在原数组上进行调整。。。

首先，我们需要知道一点：数组上的每个元素 a[i] 调整后的位置为 $(i\ast 2)\%(len+1)$，若我们使用这个公式，会发现有多个环出现，比如：

调整前的数字：a b c d e f

调整前的位置：1 2 3 4 5 6

如果从 1 位置出发，按照上述位置公式来调整，则：a(1) 推出 b(2)，b(2) 推出 d(4)，d(4) 回到 1 位置，形成一个 1->2->4->1 的环。但是此时数组还有需要调整的数字，从 3 位置出发：c(3) 推出 f(6) ，f(6) 推出 e(5) ，e(5) 回到位置 3 ，形成 3->6->5->3 这条环。

数组长度不同，环的数量不同，并且任何一条环一定不和其他的环共享任何数字，所以我们可以通过找到数组上的每个环起点，然后交换即可。

不过如何记录某个位置的元素已经被调整过了呢？肯定不能额外开一个数组记录，我们可以这样做：已经调整的元素均减去一个无穷小，保证其小于等于0，而剩下的大于0的元素肯定是没调整过的，继续调整即可。

书上的解法更高级，其跳过记录已经调整过的元素这一思路，不过我的解法更简单粗暴哈哈哈哈。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 100000;
const int INF = 1e8 + 10;

int n;
int a[N];

void solve() {
    for ( int i = 1; i <= n; ++i ) {
        if ( a[i] > 0 ) {
            int nxt = ( i << 1 ) % ( n + 1);
            int pre = a[i] - INF;
            while ( nxt != i ) {
                int tmp = a[nxt] - INF;
                a[nxt] = pre;
                pre = tmp;
                nxt = ( nxt << 1 ) % ( n + 1 );
            }
            a[nxt] = pre;
        }
    }
}

int main(void) {
    scanf("%d", &n);
    for ( int i = 1; i <= n; ++i )
        scanf("%d", a + i);
    solve();
    for ( int i = 1; i <= n; ++i )
        printf("%d%c", a[i] + INF, " \n"[i == n]);
    return 0;
}