我們正在玩一個猜數字遊戲。 遊戲規則如下:
我從 1 到 n 選擇一個數字。 你需要猜我選擇了哪個數字。
每次你猜錯了,我會告訴你這個數字是大了還是小了。
你調用一個預先定義好的接口 guess(int num),它會返回 3 個可能的結果(-1,1 或 0):
-1 : 我的數字比較小
1 : 我的數字比較大
0 : 恭喜!你猜對了!
示例 :
輸入: n = 10, pick = 6
輸出: 6
很明顯一開始想到的是二分法進行,但是不自覺就會想到數組,其實應該mid應該是 int mid=low+(high-low)/2,因爲只有這樣纔會不溢出;需要特別注意一下;
public class Solution extends GuessGame {
public int guessNumber(int n) {
int low=0,high=n;
while(low<=high){
int mid=low+(high-low)/2;
if(guess(mid)==-1) high=mid-1;
else if(guess(mid)==1) low=mid+1;
else return mid;
}
return 0;
}
}
把mid計算錯了,因此一直是超時的節湊,哎。。。。
/* The guess API is defined in the parent class GuessGame.
@param num, your guess
@return -1 if my number is lower, 1 if my number is higher, otherwise return 0
int guess(int num); */
public class Solution extends GuessGame {
public int guessNumber(int n) {
if(guess(1)==0){
return 1;
}
if(guess(n)==0){
return n;
}
int start = 0;
int end = n;
int mid = (0+n)/2;
while(guess(mid) != 0){
if(guess(end) == 0){
return end;
}
if(guess(start) == 0){
return start;
}
if(guess(mid) == 1){
start = mid + 1;
}
if(guess(mid) == -1){
end = mid - 1;
}
mid = start+(end - start)/2;
}
return mid;
}
}