每日一题(2022-04-19)——字符的最短距离

传说之下的花儿

发布于 2023-04-16 15:20:52

1040

发布于 2023-04-16 15:20:52

821. 字符的最短距离

题目描述：

给你一个字符串 s 和一个字符 c ，且 c 是 s 中出现过的字符。返回一个整数数组 answer ，其中 answer.length == s.length 且 answer[i] 是 s 中从下标 i 到离它最近的字符 c 的距离。两个下标 i 和 j 之间的距离为 abs(i - j) ，其中 abs 是绝对值函数。
示例1: 输入：s = “loveleetcode”, c = “e” 输出：[3,2,1,0,1,0,0,1,2,2,1,0] 示例2: 输入：s = “aaab”, c = “b” 输出：[3,2,1,0]

思路：

先找出字符串s中所有指定字符，并使用一个切片存储其位置，开始遍历字符串s，如果是指定字符，向结果切片追加0，否则，遍历所有指定字符，求当前字符到只有指定字符的位置，并取最小值，并追加到结果切片。

题解：

func shortestToChar(s string, c byte) []int {
	ans := make([]int, 0)
	// 存放指定字符 第几个:下标
	cSlice := make([]int, 0)
	for i, ch := range s {
		if byte(ch) == c {
			cSlice = append(cSlice, i)
		}
	}
	for i, ch := range s {
		// 遍历到指定字符，追加0
		if byte(ch) == c {
			ans = append(ans, 0)
		} else {
			dis := math.MaxFloat64
			// 求当前位置到每一个指定字符的位置，取最小
			for _, index := range cSlice {
				dis = math.Min(dis,math.Abs(float64(i-index)))
			}
			ans = append(ans, int(dis))
		}
	}
	return ans
}