zhangdizhangdi

374.猜数字大小

题目

LeetCode 简单

猜数字游戏的规则如下:
每轮游戏,我都会从 1 到 n 随机选择一个数字。 请你猜选出的是哪个数字。
如果你猜错了,我会告诉你,你猜测的数字比我选出的数字是大了还是小了。

你可以通过调用一个预先定义好的接口 int guess(int num) 来获取猜测结果,返回值一共有 3 种可能的情况(-1,1 或 0):

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

返回我选出的数字。

题解

二分 O(logn) O(1)

js
/**
 * Forward declaration of guess API.
 * @param {number} num   your guess
 * @return 	     -1 if num is higher than the picked number
 *			      1 if num is lower than the picked number
 *               otherwise return 0
 * var guess = function(num) {}
 */

const guess = function (num) {
  const picked = 4
  if (num > picked) {
    return -1
  } else if (num < picked) {
    return 1
  } else {
    return 0
  }
}

function guessNumber(n) {
  let left = 1
  let right = n

  while (left <= right) {
    const mid = left + Math.round((right - left) / 2)
    const res = guess(mid)

    console.log(mid, res)

    if (res === 1) {
      left = mid + 1
    } else if (res === -1) {
      right = mid - 1
    } else {
      return mid
    }
  }
}

console.log('🌰', guessNumber(10))
执行结果
6 -1
3 1
5 -1
4 0
🌰 4