leetcode 52 : Guess Number Higher or Lower
題目鏈接: https://leetcode.com/problems/guess-number-higher-or-lower/
難度: 簡單
歸類 : 數組操作, 二分查找
最開始刷easy題比較多,畢竟循序漸進… 二分的easy題快沒了哈哈哈
題目:
我們正在玩一個猜數字遊戲。 遊戲規則如下:
我從 1 到 n 選擇一個數字。 你需要猜我選擇了哪個數字。
每次你猜錯了,我會告訴你這個數字是大了還是小了。
你調用一個預先定義好的接口 guess(int num),它會返回 3 個可能的結果(-1,1 或 0):
-1 : 我的數字比較小
1 : 我的數字比較大
0 : 恭喜!你猜對了!
示例:
示例1:
輸入: n = 10, pick = 6
輸出: 6
解法:
主要使用c++進行了解答,以及經典題解和嘗試改進的最優/最簡潔解法。
個人解法
c++解法(set)
使用常見二分法。
//c++解法,使用二分法
/**
* Forward declaration of guess API.
* @param num your guess
* @return -1 if num is lower than the guess number
* 1 if num is higher than the guess number
* otherwise return 0
* int guess(int num);
*/
class Solution {
public:
int guessNumber(int n) {
if(n <= 1) return 1;
int start = 1;
int end = n;
int mid,curState;
while(start < end){
mid = start + (end - start) / 2;
curState = guess(mid);
if(curState == 0) return mid;
else if(curState == 1) start = mid + 1;
else end = mid - 1;
}
return start;
}
};
時間複雜度: O(logN)
空間複雜度: O(1)
提交結果:
Runtime: 0 ms, faster than 100.00% of C++ online submissions for Guess Number Higher or Lower.
Memory Usage: 5.8 MB, less than 100.00% of C++ online submissions for Guess Number Higher or Lower
題解優解
python解法(騷操作,花裏胡哨)
花裏胡哨的python解法
def guessNumber(self, n):
lo, hi = 1, n
while lo < hi:
mid = (lo + hi) / 2
lo, hi = ((mid, mid), (mid+1, hi), (lo, mid-1))[guess(mid)]
return lo
時間複雜度: O(logN)
空間複雜度: O(1)
提交結果:
Runtime: 20 ms, faster than 98.25% of Python3 online submissions for Guess Number Higher or Lower.
Memory Usage: 13.7 MB, less than 100.00% of Python3 online submissions for Guess Number Higher or Lower.
嘗試改進的最優解法
沒有太需要注意的地方。
關於mid的計算方法需要重點注意一下,有時可以看到有人使用
mid = (end - start) / 2;
而有時也會看到
mid = start + (end - start) / 2;
兩者的區別在於後者可以防止數值較大的時候導致的棧溢出。因此使用後者較爲保險。