1095. 山脉数组中查找目标值 Golang 双百 二分查找

1095. 山脉数组中查找目标值

山脉数组是先增后减数组,利用二分法找到边界。然后分别在两个单调数组里二分查找。
解题思路一看就懂,但是写代码是真的折磨,考虑边界条件头都裂了。
在这里插入图片描述

package main

import "fmt"

type MountainArray struct {
	arr []int
}

func (this *MountainArray) get(index int) int {
	return this.arr[index]
}
func (this *MountainArray) length() int {
	return len(this.arr)
}

func main() {
	m := MountainArray{[]int{1, 5, 2}}
	fmt.Println(findInMountainArray(1, &m))
}
func findInMountainArray(target int, mountainArr *MountainArray) int {
	topIndex := findMountainTop(mountainArr)
	LeftIndex := findInInc(0, topIndex+1, mountainArr, target)
	if LeftIndex >= 0 {
		return LeftIndex
	} else {
		RightIndex := findInDes(topIndex+1, mountainArr.length(), mountainArr, target)
		if RightIndex >= 0 {
			return RightIndex
		} else {
			return -1
		}
	}
	return -2
}

func findInInc(l, r int, mountainArr *MountainArray, target int) int {
	if ((target - mountainArr.get(l)) * (target - mountainArr.get(r-1))) > 0 {
		return -1
	}
	for l < r {
		m := (l + r) / 2
		if mountainArr.get(m) > target {
			r = m
		} else if mountainArr.get(m) < target {
			l = m + 1
		} else {
			return m
		}
	}
	return -1
}
func findInDes(l, r int, mountainArr *MountainArray, target int) int {
	if ((target - mountainArr.get(l)) * (target - mountainArr.get(r-1))) > 0 {
		return -1
	}
	for l < r {
		m := (l + r) / 2
		if mountainArr.get(m) > target {
			l = m + 1
		} else if mountainArr.get(m) < target {
			r = m
		} else {
			return m
		}
	}
	return -1
}
func findMountainTop(mountainArr *MountainArray) int {
	l, r := 1, mountainArr.length()-1
	for l < r-1 {
		m := (l + r) / 2
		if mountainArr.get(m) > mountainArr.get(m-1) {
			l = m
		} else {
			r = m
		}
	}
	return l
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值