內(nèi)容

給定一個字符串 S 和一個字符 C。返回一個代表字符串 S 中每個字符到字符串 S 中的字符 C 的最短距離的數(shù)組。

示例 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;
};

回到目錄