• 一、题目
  • 二、解题思路
  • 三、解题代码

    一、题目

    统计一个数字:在排序数组中出现的次数。

    举例说明

    例如输入排序数组{ 1, 2, 3, 3, 3, 3, 4, 5}和数字3 ,由于3 在这个数组中出现了4 次,因此输出4 。

    二、解题思路

    利用改进的二分算法。

    如何用二分查找算法在数组中找到第一个k,二分查找算法总是先拿数组中间的数字和k作比较。如果中间的数字比k大,那么k只有可能出现在数组的前半段,下一轮我们只在数组的前半段查找就可以了。如果中间的数字比k小,那么k只有可能出现在数组的后半段,下一轮我们只在数组的后半乓查找就可以了。如果中间的数字和k 相等呢?我们先判断这个数字是不是第一个k。如果位于中间数字的前面一个数字不是k,此时中间的数字刚好就是第一个k。如果中间数字的前面一个数字也是k,也就是说第一个k肯定在数组的前半段, 下一轮我们仍然需要在数组的前半段查找。

    同样的思路在排序数组中找到最后一个k。如果中间数字比k大,那么k只能出现在数组的前半段。如果中间数字比k小,k就只能出现在数组的后半段。如果中间数字等于k呢?我们需要判断这个k是不是最后一个k,也就是中间数字的下一个数字是不是也等于k。如果下一个数字不是k,则中间数字就是最后一个k了:否则下一轮我们还是要在数组的后半段中去查找。

    三、解题代码

    1. public class Test {
    2. /**
    3. * 找排序数组中k第一次出现的位置
    4. *
    5. * @param data
    6. * @param k
    7. * @param start
    8. * @param end
    9. * @return
    10. */
    11. private static int getFirstK(int[] data, int k, int start, int end) {
    12. if (data == null || data.length < 1 || start > end) {
    13. return -1;
    14. }
    15. int midIdx = start + (end - start) / 2;
    16. int midData = data[midIdx];
    17. if (midData == k) {
    18. if (midIdx > 0 && data[midIdx - 1] != k || midIdx == 0) {
    19. return midIdx;
    20. } else {
    21. end = midIdx - 1;
    22. }
    23. } else if (midData > k) {
    24. end = midIdx - 1;
    25. } else {
    26. start = midIdx + 1;
    27. }
    28. return getFirstK(data, k, start, end);
    29. }
    30. /**
    31. * 找排序数组中k最后一次出现的位置
    32. *
    33. * @param data
    34. * @param k
    35. * @param start
    36. * @param end
    37. * @return
    38. */
    39. private static int getLastK(int[] data, int k, int start, int end) {
    40. if (data == null || data.length < 1 || start > end) {
    41. return -1;
    42. }
    43. int midIdx = start + (end - start) / 2;
    44. int midData = data[midIdx];
    45. if (midData == k) {
    46. if (midIdx + 1 < data.length && data[midIdx + 1] != k || midIdx == data.length - 1) {
    47. return midIdx;
    48. } else {
    49. start = midIdx + 1;
    50. }
    51. } else if (midData < k) {
    52. start = midIdx + 1;
    53. } else {
    54. end = midIdx - 1;
    55. }
    56. return getLastK(data, k, start, end);
    57. }
    58. /**
    59. * 题目:统计一个数字:在排序数组中出现的次数
    60. * @param data
    61. * @param k
    62. * @return
    63. */
    64. public static int getNumberOfK(int[] data, int k) {
    65. int number = 0;
    66. if (data != null && data.length > 0) {
    67. int first = getFirstK(data, k, 0, data.length - 1);
    68. int last = getLastK(data, k, 0, data.length - 1);
    69. if (first > -1 && last > -1) {
    70. number = last - first + 1;
    71. }
    72. }
    73. return number;
    74. }
    75. }