关于golang:Golang力扣Leetcode-374-猜数字大小二分查找

47次阅读

共计 874 个字符,预计需要花费 3 分钟才能阅读完成。

题目 :猜数字游戏的规定如下:

  • 每轮游戏,我都会从 1n 随机抉择一个数字。请你猜选出的是哪个数字。
  • 如果你猜错了,我会通知你,你猜想的数字比我选出的数字是大了还是小了。

你能够通过调用一个事后定义好的接口 int guess(int num) 来获取猜想后果,返回值一共有 3 种可能的状况(-1,1 或 0):

  • -1:我选出的数字比你猜的数字小 pick < num
  • 1:我选出的数字比你猜的数字大 pick > num
  • 0:我选出的数字和你猜的数字一样。祝贺!你猜对了!pick == num
    返回我选出的数字。

链接 :力扣 Leetcode – 374. 猜数字大小.

示例 1:

输出:n = 10, pick = 6
输入:6

示例 2:

输出:n = 1, pick = 1
输入:1

示例 3:

输出:n = 2, pick = 1
输入:1

示例 4:

输出:n = 2, pick = 2
输入:2

思路 :用二分查找,mid = left + right

  • 如果 guess(mid) = -1,证实咱们猜的数比他想的大,right = mid – 1 放大范畴;
  • 如果 guess(mid) = 1,证实咱们猜的数比他想的小,left = mid + 1 放大范畴;
  • 如果 guess(mid) = 0,就证实他想的数就是 mid

直至找到他要的数 return 进去

Go 代码:

package main

import "fmt"

func guess(num int) int {
    // 1~10 假如我想的数是 6
    if num < 6 {
        return 1
        // 1:我选出的数字比你猜的数字大
    } else if num > 6 {
        return -1
        // -1:我选出的数字比你猜的数字小
    } else {return 0}
}

func guessNumber(n int) int {
    // 二分查找
    left, right := 1, n
    for left <= right {mid := (left + right) / 2
        guess := guess(mid)
        if guess == -1 {right = mid - 1} else if guess == 1 {left = mid + 1} else if guess == 0 {return mid}
    }
    return left
}

func main() {fmt.Println(guessNumber(10))
}

提交截图

正文完
 0