內(nèi)容
給定一個(gè)字符串 S 和一個(gè)字符 C。返回一個(gè)代表字符串 S 中每個(gè)字符到字符串 S 中的字符 C 的最短距離的數(shù)組。
示例 1:
輸入: S = "loveleetcode", C = 'e'
輸出: [3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]
說明:
字符串 S 的長度范圍為 [1, 10000]。
C 是一個(gè)單字符,且保證是字符串 S 里的字符。
S 和 C 中的所有字母均為小寫字母。
思路
大致思路就是,先將C的位置都找出來,然后每次取兩個(gè)位置j,k,從0開始往中間加,直到i=j,然后對(duì)于位置集合最左邊和最右邊,則分別遞增處理。
代碼
/**
大致思路就是,先將C的位置都找出來,然后每次取兩個(gè)位置j,k,從0開始往中間加,直到i=j,然后對(duì)于位置集合最左邊和最右邊,則分別遞增處理。
* @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;
};