smallest-range.go

package main

import (
	"fmt"
)

// MinMax comment
func MinMax(A []int) (int, int) {
	min := A[0]
	max := A[0]
	for _, e := range A {
		if e < min {
			min = e
		}
		if e > max {
			max = e
		}
	}
	return min, max
}

func smallestRangeI(A []int, K int) int {
	min, max := MinMax(A)
	temBe, temHi, distant := 0, 0, max-min
	for i := 1; i <= K; i++ {
		temBe = min + i
		temHi = max - i
		temDis := temHi - temBe
		if temDis >= 0 && temDis < distant {
			distant = temDis
		}
		if (i + 1) <= K {
			temBe = min + i
			temHi = max - i - 1
			temDis = temHi - temBe
			if temDis >= 0 && temDis < distant {
				distant = temDis
			}
		}

		temBe = min + i - 1
		temHi = max - i
		temDis = temHi - temBe
		if temDis >= 0 && temDis < distant {
			distant = temDis
		}
	}

	return distant
}

func main() {
	a := []int{7, 8, 8}
	k := 5
	res := smallestRangeI(a, k)
	fmt.Println("res:", res)
}