2020-11-11:手写代码:如何获得有序数组中指定元素的个数?

举报
福大大架构师每日一题 发表于 2020/11/11 21:11:24 2020/11/11
【摘要】 福哥答案2020-11-11:1.遍历法。无代码。2.二分法。二分查找元素,然后二分查找左边界,再查找右边界,最后右边界减去左边界就是指定元素个数。这道题实际上是如下三道题的综合。1) 在一个有序数组中,找某个数是否存在 。2) 在一个有序数组中,找>=某个数最左侧的位置 。3) 在一个有序数组中,找<=某个数最右侧的位置 。golang代码如下:package mainimport "fm...

福哥答案2020-11-11:


1.遍历法。无代码。

2.二分法。二分查找元素,然后二分查找左边界,再查找右边界,最后右边界减去左边界就是指定元素个数。这道题实际上是如下三道题的综合。

1) 在一个有序数组中,找某个数是否存在 。

2) 在一个有序数组中,找>=某个数最左侧的位置 。

3) 在一个有序数组中,找<=某个数最右侧的位置 。


golang代码如下:

package main

import "fmt"

func main() {
    arr := []int{0, 1, 2, 3, 3, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 7, 8}
    v := 3
    fmt.Println(v, "的个数是:", BSCount(arr, v))
}

//二分法
func BSCount(arr []int, v int) int {
    L := 0
    R := len(arr) - 1
    M := -1
    // L..R

    mid := -1
    //找目标值并且缩小左边界L和右边界R的范围
    for L <= R {
        mid = L + (R-L)>>1
        if arr[mid] == v {
            M = mid
            break
        } else if arr[mid] > v {
            R = mid - 1
        } else {
            L = mid + 1
        }
    }
    //没找到目标值,直接返回
    if M == -1 {
        return 0
    }

    index := 0
    LL := L //缓存原始左边界
    RR := R //缓存原始右边界

    //找左边界
    R = M //缩小范围
    for L <= R {
        mid = L + (R-L)>>1
        if arr[mid] >= v {
            index = mid
            R = mid - 1
        } else {
            L = mid + 1
        }
    }
    LL = index //左边界确定下来了
    R = RR     //原始右边界已经发生变化,需要恢复到以前的边界

    //找右边界
    L = M //缩小范围
    for L <= R {
        mid = L + (R-L)>>1
        if arr[mid] <= v {
            index = mid
            L = mid + 1
        } else {
            R = mid - 1
        }
    }
    RR = index //右边界确定下来了

    return RR - LL + 1
}

执行结果如下:

image.png

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。