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,否则,遍历所有指定字符,求当前字符到只有指定字符的位置,并取最小值,并追加到结果切片。
题解:
代码语言:javascript复制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
}