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 }
提交结果: