First Bad Version

Question

  1. The code base version is an integer and start from 1 to n. One day, someone commit a bad version in the code case, so it caused itself and the following versions are all failed in the unit tests.
  2. You can determine whether a version is bad by the following interface:
  4. Java:
  5.     public VersionControl {
  6.         boolean isBadVersion(int version);
  7.     }
  8. C++:
  9.     class VersionControl {
  10.     public:
  11.         bool isBadVersion(int version);
  12.     };
  13. Python:
  14.     class VersionControl:
  15.         def isBadVersion(version)
  17. Find the first bad version.
  18. Note
  19. You should call isBadVersion as few as possible.
  21. Please read the annotation in code area to get the correct way to call isBadVersion in different language. For example, Java is VersionControl.isBadVersion.
  23. Example
  24. Given n=5
  26. Call isBadVersion(3), get false
  28. Call isBadVersion(5), get true
  30. Call isBadVersion(4), get true
  32. return 4 is the first bad version
  34. Challenge
  35. Do not call isBadVersion exceed O(logn) times.

題 Search for a Range 的變形，找出左邊界即可。

Java

  1. /**
  2.  * public class VersionControl {
  3.  *     public static boolean isBadVersion(int k);
  4.  * }
  5.  * you can use VersionControl.isBadVersion(k) to judge wether 
  6.  * the kth code version is bad or not.
  7. */
  8. class Solution {
  9.     /**
  10.      * @param n: An integers.
  11.      * @return: An integer which is the first bad version.
  12.      */
  13.     public int findFirstBadVersion(int n) {
  14.         // write your code here
  15.         if (n == 0) {
  16.             return -1;
  17.         }
  19.         int start = 1, end = n, mid;
  20.         while (start + 1 < end) {
  21.             mid = start + (end - start)/2;
  22.             if (VersionControl.isBadVersion(mid) == false) {
  23.                 start = mid;
  24.             } else {
  25.                 end = mid;
  26.             }
  27.         }
  29.         if (VersionControl.isBadVersion(start) == true) {
  30.             return start;
  31.         } else if (VersionControl.isBadVersion(end) == true) {
  32.             return end;
  33.         } else {
  34.             return -1; // not found
  35.         }
  36.     }
  37. }

C++

  1. /**
  2.  * class VersionControl {
  3.  *     public:
  4.  *     static bool isBadVersion(int k);
  5.  * }
  6.  * you can use VersionControl::isBadVersion(k) to judge wether
  7.  * the kth code version is bad or not.
  8. */
  9. class Solution {
  10. public:
  11.     /**
  12.      * @param n: An integers.
  13.      * @return: An integer which is the first bad version.
  14.      */
  15.     int findFirstBadVersion(int n) {
  16.         if (n < 1) {
  17.             return -1;
  18.         }
  20.         int start = 1;
  21.         int end = n;
  22.         int mid;
  23.         while (start + 1 < end) {
  24.             mid = start + (end - start) / 2;
  25.             if (VersionControl::isBadVersion(mid)) {
  26.                 end = mid;
  27.             } else {
  28.                 start = mid;
  29.             }
  30.         }
  32.         if (VersionControl::isBadVersion(start)) {
  33.             return start;
  34.         } else if (VersionControl::isBadVersion(end)) {
  35.             return end;
  36.         }
  38.         return -1;  // find no bad version
  39.     }
  40. };

源碼分析

找左邊界和Search for a Range類似，但是最好要考慮到有可能end處也為good version，此部分異常也可放在開始的時候處理。

Python

  1. #class VersionControl:
  2. #    @classmethod
  3. #    def isBadVersion(cls, id)
  4. #        # Run unit tests to check whether verison `id` is a bad version
  5. #        # return true if unit tests passed else false.
  6. # You can use VersionControl.isBadVersion(10) to check whether version 10 is a
  7. # bad version.
  8. class Solution:
  9.     """
  10.     @param n: An integers.
  11.     @return: An integer which is the first bad version.
  12.     """
  13.     def findFirstBadVersion(self, n):
  14.         if n < 1:
  15.             return -1
  17.         start, end = 1, n
  18.         while start + 1 < end:
  19.             mid = start + (end - start) / 2
  20.             if VersionControl.isBadVersion(mid):
  21.                 end = mid
  22.             else:
  23.                 start = mid
  25.         if VersionControl.isBadVersion(start):
  26.             return start
  27.         elif VersionControl.isBadVersion(end):
  28.             return end
  30.         return -1

Leetcode版題解

很明顯使用二分搜索，此題的測試只有Bad version必定出現的情況，會不會全部都是好的，可以向面試官詢問清楚，二分搜索的範圍，仍然使用下標範圍[0, n)控制邊界，要注意的是返回值是產品的編號，記得+1。另外直接使用產品編號[1, n+1)是行不通的，因為當n是INT_MAX時就會出現問題。

  1. // Forward declaration of isBadVersion API.
  2. bool isBadVersion(int version);
  4. class Solution {
  5. public:
  6.     int firstBadVersion(int n) {
  7.         if(isBadVersion(1)) return 1;
  8.         int lo = 0, hi = n;
  9.         while(lo < hi){
  10.             int m = lo + (hi - lo)/2;
  11.             if(!isBadVersion(m) and isBadVersion(m+1)) 
  12.                 return m+1;
  13.             else if(isBadVersion(m) and isBadVersion(m+1)) 
  14.                 hi = m;
  15.             else 
  16.                 lo = m + 1;
  17.         }
  18.     }
  19. };