go - 是 golang 的 2 的幂

Question

我想检查给定的数字是否是 2 的幂。我写了一个代码，但我无法返回 true 或 false，我认为某处存在无限循环。我只允许在代码中使用导入包中的函数。我不知道该怎么做才能纠正错误。如果你能帮助我，我会很高兴:)

package main

import (
    "os"
    "strconv"
)

func main() {
    for len(os.Args) == 2 {
        numbers, err := strconv.Atoi(os.Args[1])
        if err != nil {
            panic(err)
        }
        newnum := numbers
        counts := 0
        for numbers != 1 {
            if newnum%2 != 0 {
            } else {
                newnum = newnum / 2
            }
            counts++
        }
        var x int = 2 ^ counts
        if x == numbers {
            return true
        } else {
            return false
        }
    }
}
`

Answer 1

正如@phuclv 所评论的，我已经使用 n & (n - 1) == 0 为您的场景创建了一个示例程序，如下所示:

//Let's assume n = 16(00010000)

//Now find x = n-1 => 15(00001111) => x & n => 0

func CheckPowerOfTwo(n int) int {
    //added one corner case if n is zero it will also consider as power 2
    if n==0{
    return 1
    }
    return n & (n - 1)
}
func main() {
    var n = 16
    flag := CheckPowerOfTwo(n)
    if flag == 0 {
        fmt.Printf("Given %d number is the power of 2.\n", n)
    } else {
        fmt.Printf("Given %d number is not the power of 2.\n", n)
    }
}

你可以在这里运行它:https://go.dev/play/p/9cRWwiFAIn8