不同的猜数字游戏 — leetcode 375. Guess Number Higher or Lower II

很久没切 leetcode 的题了，静下心来切了道，这道题比较有意思，和你们分享下。

我把它叫作 "不同的猜数字游戏"，咱们先来看看传统的猜数字游戏，Guess Number Higher or Lower。题意很是的简单，给定一个数字 n，系统会随机从 1 到 n 中抽取一个数字，你须要写一个函数 guessNumber，它的做用是返回系统选择的数字，同时你还有一个外部的 API 能够调用，是为 guess 函数，它会将你猜的数字和系统选择的数字比较，是大了仍是小了。

很是的简单，稍微有点常识的童鞋应该都能想到二分查找的方案（插句题外话，这游戏让我想到了儿时的幸运52）。关于二分查找，能够参考下我之前写的一篇文章 http://www.cnblogs.com/zichi/p/5118032.html，几乎囊获了全部二分查找的状况。这道题代码比较简单，能够参考 guess-number-higher-or-lower.cpp，比较蛋疼的是不支持 JavaScript。

核心代码：

int guessNumber(int n) {
  int start = 1, end = n;
  int ans;

  while (start <= end) {
    int mid = start + (end - start) / 2;
    int val = guess(mid);

    if (val == -1)
      end = mid - 1;
    else if (val == 1)
      start = mid + 1;
    else {
      ans = mid;
      break;
    }
  }

  return ans;
}

还有一点须要注意下，取 mid 值时不能用 (start + end) / 2，否则会溢出，TLE 掉！

接着进入正题，来看 Guess Number Higher or Lower II 这道题，跟前者比，有何区别呢？一样是给定一个数字 n，系统会随机从 1 到 n 中选择一个整数，你要作的仍是将这个数猜出来。你每猜一个数字，须要花费必定的 money，好比你猜 m，那么你就要花费 m，求解你要将这个数字猜出来，至少须要的 money。

举个栗子，好比 n 为 5，系统选择的数字是 1。若是我先猜 3，系统提示你猜大了，而后再猜 2，系统提示你猜大了，那么你就能够肯定是 1 了，花费 3+2=5。可是很明显第二次猜想应该猜 1，这样花费就少了。再好比我选的是 4，第一次仍是猜 3，系统提示你猜小了，第二次猜 4，中了，总共花费 3+4=7，若是 n 为 5，至少须要 7？非也，正确的解法是先猜 4，若是数字在 1-3 之间，那么再猜 2，至少须要的应该是 6！

这是一道很典型的动态规划题，你根本不可能去盲目地猜，而后使劲地暴力递归去解！这样的复杂度是指数级的。是否可以递推求解？好比已经知道 n 为 1-5 的状况，当 n 为 6 时，第一次猜，咱们能够有 6 种猜法，分别选择 1,2,3,4,5 和 6，咱们以猜 3 为例，好比说第一把猜了 3，那么若是猜的大了，那么咱们接下去要求的是从 [1, 2] 中猜到正确数字所须要花费的最少 money，记为 x，若是猜的小了，那么咱们接下去要求的是从 [4, 6] 中猜到正确数字所须要花费的最少 money，记为 y，若是恰好猜中，则结束。很显然，若是第一把猜 3，那么猜中数字至少须要花费的 money 为 3 + max(x, y, 0)，"至少须要的花费"，就要咱们 "作最坏的打算，尽最大的努力"，即取最大值。这是第一把取 3 的状况，咱们还须要考虑其余 5 种状况，而后六种状况再取个最小值，就是 n=6 至少须要的 money！（想一想，是否是这样？）

最后来编码，咱们须要一个二维数组来表示最值。首先咱们定义一个二维数组 ans[][]，ans[i][j] 表示 i-j 中任取一个数字，猜中这个数字须要至少花费的 money。

定义 ans 数组，而且初始化：

// ans[i][j] 表示从 [i, j] 中任取一个数字
// 猜中这个数字至少须要花费的 money
var ans = [];
for (var i = 0; i <= n; i++)
  ans[i] = [];

接着咱们定义一个函数 DP，DP(ans, x, y) 表示 [x, y] 中任取一个数字，猜中这个数字须要花费的最少 money，而 ans 是为数组的引用。很显然，咱们要求的就是 DP(ans, 1, n) 的返回值，直接看代码。

function DP(ans, from, to) {
  // 若是 from >= to
  if (from >= to)
    return 0;

  // 若是 ans[from][to] 已经求得
  // 直接 return
  if (ans[from][to])
    return ans[from][to];

  // 先赋值 Infinity，便于以后的比较
  ans[from][to] = Infinity;

  // 如今要从 [from, to] 中猜数字
  // 假设先猜 i，i 能够是 [from, to] 中的任何数字，遍历之
  for (var i = from; i <= to; i++) {
    // left 为从 [from, i - 1] 猜对数字至少须要花费的 money
    var left = DP(ans, from, i - 1);
    // right 为从 [i + 1, to] 猜对数字至少须要花费的 money
    var right = DP(ans, i + 1, to);

    // tmp 为先猜 i，从 [from, to] 猜对数字至少须要花费的 money
    var tmp = i + Math.max(left, right);

    // 跟别的方案比较（即跟不是先猜 i 的方法比较）
    // 取最小值
    ans[from][to] = Math.min(ans[from][to], tmp);
  }

  return ans[from][to];
}

注释写的很清晰了，若是再细分的话，我的以为这能够说是一道 "记忆化DP"，不晓得有没有这个词？好像只据说过 "记忆化搜索"？DP 原本就是记忆化的过程吧？好了不钻牛角尖了，完整代码能够从咱们的 Repo https://github.com/hanzichi/leetcode 获取。