内容

给定一个字符串 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;
};

回到目录