前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >2021-08-03:完美洗牌问题。给定一个长度为偶数的数组arr,假设长度为N*2,左部分:arr[L1……Ln],右部分:

2021-08-03:完美洗牌问题。给定一个长度为偶数的数组arr,假设长度为N*2,左部分:arr[L1……Ln],右部分:

作者头像
福大大架构师每日一题
发布2021-08-05 16:32:47
5800
发布2021-08-05 16:32:47
举报

2021-08-03:完美洗牌问题。给定一个长度为偶数的数组arr,假设长度为N*2,左部分:arr[L1……Ln],右部分:arr[R1……Rn],请把arr调整成arr[L1,R1,L2,R2,L3,R3,…,Ln,Rn]。要求:时间复杂度O(N),额外空间复杂度O(1)。

福大大 答案2021-08-03:

这道题用自然智慧,很难想到。

i从1开始。下标循环怼。

3的k次方-1。1,3,9……是环的其中一个位置。

左部分:2*i。

右部分:(i-长/2)*2-1。

时间复杂度:O(N)。

空间复杂度:O(1)。

代码用golang编写。代码如下:

代码语言:javascript
复制
package main

import "fmt"

func main() {
    arr := []int{1, 2, 3, 4, 5, 6, 7, 8}
    shuffle(arr)
    fmt.Println(arr)
}

// 数组的长度为len,调整前的位置是i,返回调整之后的位置
// 下标不从0开始,从1开始
func modifyIndex1(i int, len2 int) int {
    if i <= len2/2 {
        return 2 * i
    } else {
        return 2*(i-(len2/2)) - 1
    }
}

// 数组的长度为len,调整前的位置是i,返回调整之后的位置
// 下标不从0开始,从1开始
func modifyIndex2(i int, len2 int) int {
    return (2 * i) % (len2 + 1)
}

// 主函数
// 数组必须不为空,且长度为偶数
func shuffle(arr []int) {
    if len(arr) != 0 && (len(arr)&1) == 0 {
        shuffle0(arr, 0, len(arr)-1)
    }
}

// 在arr[L..R]上做完美洗牌的调整(arr[L..R]范围上一定要是偶数个数字)
func shuffle0(arr []int, L int, R int) {
    for R-L+1 > 0 { // 切成一块一块的解决,每一块的长度满足(3^k)-1
        len2 := R - L + 1
        base := 3
        k := 1
        // 计算小于等于len并且是离len最近的,满足(3^k)-1的数
        // 也就是找到最大的k,满足3^k <= len+1
        for base <= (len2+1)/3 { // base > (N+1)/3
            base *= 3
            k++
        }
        // 3^k -1
        // 当前要解决长度为base-1的块,一半就是再除2
        half := (base - 1) / 2
        // [L..R]的中点位置
        mid := (L + R) / 2
        // 要旋转的左部分为[L+half...mid], 右部分为arr[mid+1..mid+half]
        // 注意在这里,arr下标是从0开始的
        rotate(arr, L+half, mid, mid+half)
        // 旋转完成后,从L开始算起,长度为base-1的部分进行下标连续推
        cycles(arr, L, base-1, k)
        // 解决了前base-1的部分,剩下的部分继续处理
        L = L + base - 1 // L ->     [] [+1...R]
    }
}

// 从start位置开始,往右len的长度这一段,做下标连续推
// 出发位置依次为1,3,9...
func cycles(arr []int, start int, len2 int, k int) {
    // 找到每一个出发位置trigger,一共k个
    // 每一个trigger都进行下标连续推
    // 出发位置是从1开始算的,而数组下标是从0开始算的。
    for i, trigger := 0, 1; i < k; i, trigger = i+1, trigger*3 {
        preValue := arr[trigger+start-1]
        cur := modifyIndex2(trigger, len2)
        for cur != trigger {
            tmp := arr[cur+start-1]
            arr[cur+start-1] = preValue
            preValue = tmp
            cur = modifyIndex2(cur, len2)
        }
        arr[cur+start-1] = preValue
    }
}

// [L..M]为左部分,[M+1..R]为右部分,左右两部分互换
func rotate(arr []int, L int, M int, R int) {
    reverse(arr, L, M)
    reverse(arr, M+1, R)
    reverse(arr, L, R)
}

// [L..R]做逆序调整
func reverse(arr []int, L int, R int) {
    for L < R {
        arr[L], arr[R] = arr[R], arr[L]
        L++
        R--
    }
}

执行结果如下:

***

[左神java代码](https://github.com/algorithmzuo/coding-for-great-offer/blob/main/src/class20/Code03_ShuffleProblem.java)

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2021-08-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 福大大架构师每日一题 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档