有人可以评论这是否是在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
}
}
答案 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
}