题目描述:
给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。
示例 1:
输入: S = “loveleetcode”, C = ‘e’
输出: [3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]
说明:
字符串 S 的长度范围为 [1, 10000]。
C 是一个单字符,且保证是字符串 S 里的字符。
S 和 C 中的所有字母均为小写字母。
我的代码
class Solution {
public int[] shortestToChar(String S, char C) {
int [] result = new int[S.length()];
List<Integer> temMap = new ArrayList<>();
for (int i = 0; i < S.length(); i++) {
char tem = S.charAt(i);
if(tem == C){
temMap.add(i);
result[i] = 0;
}
}
int index = 0;
for (int i = 0; i < S.length(); i++) {
if(S.charAt(i) != C){
if(index == 0){
result[i] = Math.abs(temMap.get(index) - i);
}else {
if(temMap.size() == index){
result[i] = Math.abs(temMap.get(index - 1) - i);
}else
result[i] = Math.min(Math.abs(temMap.get(index -1) - i),
Math.abs(temMap.get(index) - i));
}
}else {
index ++;
}
}
return result;
}
}