这是一个合理而惯用的GoLang循环移位实施吗?

时间:2015-10-10 22:04:59

标签: go

有人可以评论这是否是在Go中实现circular shift整数数组的合理且惯用的方式? (我故意选择不使用按位操作。)

怎么可以改进?

package main

import "fmt"

func main() {
    a := []int{1,2,3,4,5,6,7,8,9,10}
    fmt.Println(a)
    rotateR(a, 5)
    fmt.Println(a)
    rotateL(a, 5)
    fmt.Println(a)
}

func rotateL(a []int, i int) {
    for count := 1; count <= i; count++ {
        tmp := a[0]
        for n := 1;n < len(a);n++ {
            a[n-1] = a[n]
        }
        a[len(a)-1] = tmp
    }
}

func rotateR(a []int, i int) {
    for count := 1; count <= i; count++ {
        tmp := a[len(a)-1]
        for n := len(a)-2;n >=0 ;n-- {
            a[n+1] = a[n]
        }
        a[0] = tmp
    }
}

3 个答案:

答案 0 :(得分:4)

一次旋转切片一个位置,并重复以获得所需的总旋转意味着需要时间与旋转距离×切片长度成比例。通过将每个元素直接移动到其最终位置,您可以在与切片长度成比例的时间内完成此操作。

The code for this比你有点棘手,你需要一个GCD函数来确定切片的次数:

func gcd(a, b int) int {
    for b != 0 {
        a, b = b, a % b
    }

    return a
}

func rotateL(a []int, i int) {

    // Ensure the shift amount is less than the length of the array,
    // and that it is positive.
    i = i % len(a)
    if i < 0 {
        i += len(a)
    }

    for c := 0; c < gcd(i, len(a)); c++ {

        t := a[c]

        j := c

        for {
            k := j + i
            // loop around if we go past the end of the slice
            if k >= len(a) {
                k -= len(a)
            }
            // end when we get to where we started
            if k == c {
                break
            }
            // move the element directly into its final position
            a[j] = a[k]
            j = k
        }

        a[j] = t
    }
}

通过 p 位置旋转 l 大小的切片相当于将旋转 l - p 位置,因此您可以使用rotateR简化rotateL功能:

func rotateR(a []int, i int) {
    rotateL(a, len(a) - i)
}

答案 1 :(得分:2)

您的代码适用于就地修改。

不清楚理解按位操作的含义。也许这个

package main

    import "fmt"

    func main() {
        a := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
        fmt.Println(a)
        rotateR(&a, 4)
        fmt.Println(a)
        rotateL(&a, 4)
        fmt.Println(a)
    }

    func rotateL(a *[]int, i int) {
        x, b := (*a)[:i], (*a)[i:]
        *a = append(b, x...)
    }

    func rotateR(a *[]int, i int) {
        x, b := (*a)[:(len(*a)-i)], (*a)[(len(*a)-i):]
        *a = append(b, x...)
    }

代码有效https://play.golang.org/p/0VtiRFQVl7

在Go词汇表中称为重新复制。权衡是在你的代码片段与动态分配中应对和循环。这是你的选择,但是如果将10000个元素阵列移位一个位置,那么复制会更便宜。

答案 2 :(得分:0)

我喜欢Uvelichitel解决方案,但如果您想使用O(n)复杂度的模块化算法

package main
func main(){
   s := []string{"1", "2", "3"}
   rot := 5
   fmt.Println("Before RotL", s)
   fmt.Println("After RotL", rotL(rot, s))
   fmt.Println("Before RotR", s)
   fmt.Println("After RotR", rotR(rot,s))

}
func rotL(m int, arr []string) []string{
     newArr := make([]string, len(arr))

     for i, k := range arr{
         newPos := (((i - m) % len(arr)) + len(arr)) % len(arr)
         newArr[newPos] = k
     }

     return newArr
}

func rotR(m int, arr []string) []string{

     newArr := make([]string, len(arr))

     for i, k := range arr{
         newPos := (i + m) % len(arr)
         newArr[newPos] = k
      }
      return newArr
}