53. 数字在排序数组中出现的次数

题目链接

牛客网 (opens new window)

题目描述

Input:
nums = 1, 2, 3, 3, 3, 3, 4, 6
K = 3

Output:
4

解题思路

package main

import (
    "sort"
)

/**
 * 
 * @param data int整型一维数组 
 * @param k int整型 
 * @return int整型
*/
func GetNumberOfK( data []int ,  k int ) int {
    // write code here
    if len(data) == 0 {
        return 0
    }
    
    index := sort.SearchInts(data, k)
    if index == len(data) {
        return 0
    }
    
    total := 0
    for i := index; i < len(data); i ++ {
        if data[i] == k {
            total ++
        } else {
            break
        }
    }
    for i := index-1; i >= 0; i -- {
        if data[i] == k {
            total ++
        } else {
            break
        }
    }
    
    return total
}