題目描述
統計一個數字在排序數組中出現的次數。java
方法一
最簡單的方法就是遍歷整個數組。沒什麼好說的,很low.面試
/** * 方法一 * @param array * @param k * @return */ public int GetNumberOfK(int [] array , int k) { if(array == null || array.length == 0) { return 0; } int times = 0; for (int i = 0; i < array.length; i++) { if(array[i] == k) { ++times; } } return times; }
方法二
數組是有序的,咱們能夠利用二分查找。直觀的一個思路就是經過二分查找得到一個k在數組中的下標,而後順序的在k的前面和後面尋找和k相等的數。數組
/** * 方法二 */ public int GetNumberOfK_2(int [] array , int k) { if(array == null || array.length == 0) { return 0; } int low = 0, high = array.length - 1, mid = 0, times = 0; while(low <= high) { mid = (low + high) / 2; if(k < array[mid]) { high = mid - 1; } else if(k > array[mid]) { low = mid + 1; } else { times = 1; break; } } for (low = mid - 1; (low >= 0) && (array[low] == k); low--) { times += 1; } for (low = mid + 1; (low < array.length) && (array[low] == k); low++) { times += 1; } return times; }
可是極可能面試官還不滿意,還想要更快的方法。spa
方法三
經過二分查找得到數組中第一個k的下標和最後一個k的下標,而後下標相減就是出現的次數。思路是經過二分找到一個k,先判斷這個數字是否是第一個k,若是中間數字的前面一個數字不是k,那麼此時中間數字恰好就是第一個k,若是中間數字的前面一個數字也是k,那麼第一個k確定在數組的前半段,繼續二分,這是遞歸的過程。
尋找第一個k下標的方法(遞歸):code
private int FetFirstK(int[] elem, int k, int start, int end) { if(start > end) { return -1; } int mid = (start + end) / 2; if(elem[mid] > k) { return FetFirstK(elem, k, start, mid - 1); } else if(elem[mid] < k) { return FetFirstK(elem, k, mid + 1, end); } else if(mid - 1 >= 0 && elem[mid - 1] == k) { return FetFirstK(elem, k, start, mid - 1); } else { return mid; } }
尋找最後一個k的下標的方法(非遞歸):排序
private int FetLastK(int[] elem, int k, int start, int end) { int mid = (start + end) / 2; while(start <= end) { if(elem[mid] > k) { end = mid - 1; } else if(elem[mid] < k) { start = mid + 1; } else if(mid + 1 <elem.length && elem[mid + 1] == k) { start = mid + 1; } else { return mid; } mid = (start + end) / 2; } return -1; }
主方法:遞歸
/** * 方法三 * @param array * @param k * @return */ public int GetNumberOfK_3(int [] array , int k) { if(array == null || array.length == 0) { return 0; } int firstK = FetFirstK(array, k, 0, array.length - 1); int lastK = FetLastK(array, k, 0, array.length - 1); if(firstK != -1 && lastK != -1) { return lastK - firstK + 1; } return 0; }
方法四
由於數組中都是整數,因此能夠稍微變一下,不是搜索k的兩個位置,而是搜索k-0.5和k+0.5這兩個數應該插入的位置,而後相減便可。ast
/** * 方法四 * @param array * @param k * @return */ public int GetNumberOfK_4(int [] array , int k) { if(array == null || array.length == 0) { return 0; } return biSearch(array, k + 0.5) - biSearch(array, k - 0.5); } private int biSearch(int[] elem, double num) { int start = 0, end = elem.length - 1, mid = (start + end) / 2; while(start <= end) { if(elem[mid] < num) { start = mid + 1; } else if(elem[mid] > num) { end = mid - 1; } mid = (start + end) / 2; } return start; }