数组09--数字在升序数组中出现的次数-CFANZ编程社区

数组09--数字在升序数组中出现的次数-jz37

题目概述
解析&参考答案
注意事项
说明

题目概述

算法说明
统计一个数字在升序数组中出现的次数。
测试用例
输入：
[1,2,3,3,3,3,4,5],3
输出：
4

解析&参考答案

解析
方法1：扫描一般，记录出现的次数，时间复杂度过高；
方法2：使用二分查找提高时间复杂度，分别查找最左边的的位置L和最右边的位置R，则结果为R-L+1。
参考答案

vim jz37.go
package main

import "fmt"

func GetNumberOfKV2(data []int, k int) int {
    ret := 0
    if data == nil || len(data) == 0 {
        return 0
    }
    for _, v := range data {
        if v == k {
            ret++
        }
    }
    return ret
}

func BinarySearchLeftK(data []int, k, start, end int) int {
    if start > end {
        return -1
    }
    mid := (start + end) / 2
    if data[mid] > k {
        return BinarySearchLeftK(data, k, start, mid-1)
    } else if data[mid] < k {
        return BinarySearchLeftK(data, k, mid+1, end)
    } else {
        // 找到k后，向左移动以便于确认左边相邻的是否为K
        cur := mid
        for i := mid - 1; i >= start; i-- {
            if data[i] == k {
                cur = i
            } else {
                break
            }
        }
        return cur
    }
}

func BinarySearchRightK(data []int, k, start, end int) int {
    if start > end {
        return -1
    }
    mid := (start + end) / 2
    if data[mid] > k {
        return BinarySearchRightK(data, k, start, mid-1)
    } else if data[mid] < k {
        return BinarySearchRightK(data, k, mid+1, end)
    } else {
        // 找到k后，向右移动以便于确认右边相邻的是否为K
        cur := mid
        for i := mid + 1; i <= end; i++ {
            if data[i] == k {
                cur = i
            } else {
                break
            }
        }
        return cur
    }
}

func GetNumberOfK(data []int, k int) int {
    if data == nil || len(data) == 0 {
        return 0
    }
    left := BinarySearchLeftK(data, k, 0, len(data)-1)
    right := BinarySearchRightK(data, k, 0, len(data)-1)
    if left != -1 && right != -1 {
        return right - left + 1
    } else {
        return 0
    }
}

func main() {
    array := []int{1, 2, 3, 3, 3, 3, 4, 5}
    k := 3
    // result := GetNumberOfKV2(array, k)
    result := GetNumberOfK(array, k)
    fmt.Println(result)
}