内容
给定一个字符串 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 中的所有字母均为小写字母。
思路
大致思路就是,先将C的位置都找出来,然后每次取两个位置j,k,从0开始往中间加,直到i=j,然后对于位置集合最左边和最右边,则分别递增处理。
代码
/**
大致思路就是,先将C的位置都找出来,然后每次取两个位置j,k,从0开始往中间加,直到i=j,然后对于位置集合最左边和最右边,则分别递增处理。
* @param {string} S
* @param {character} C
* @return {number[]}
*/
var shortestToChar = function (S, C) {
var array = new Array(S.length);
var positions = [];
for (var i = 0; i < S.length; i++) {
if (S[i] == C) {
array[i] = 0;
positions.push(i);
}
}
for (var i = 0; i < positions.length - 1; i++) {
var start = positions[i];
var end = positions[i + 1];
for (var j = start, k = end; j <= k; j++, k--) {
array[j] = j - start;
array[k] = end - k;
}
}
for (var i = positions[0]; i >= 0; i--) {
array[i] = positions[0] - i;
}
for (var i = positions.pop(), start = i; i < array.length; i++) {
array[i] = i - start;
}
return array;
};
网友评论