744. 寻找比目标字母大的最小字母 golang

官解

func nextGreatestLetter(letters []byte, target byte) byte {
	idx := 0
	for i := 0; i < len(letters); i++ {
		if letters[i] > target {
			idx = i
			break
		}
	}
	return letters[idx]
}

Me

这个题答得很差劲，可能有点累

func nextGreatestLetter(letters []byte, target byte) byte {
	var i int = 0
	max := letters[0]

	for i != len(letters) && target >= letters[i] {
		i++
		if i == len(letters) {
			return letters[0]
		}
		max = letters[i]
	}
	if i >= len(letters) {
		return letters[0]
	}
	return max
}