165、字符的最短距离

题目描述:
给定一个字符串 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;
		 
    }
}

猜你喜欢

转载自blog.csdn.net/qq_34446716/article/details/85854948