[LeetCode] 821. Shortest Distance to a Character

Given a string s and a character c that occurs in s, return an array of integers answer where answer.length == s.length and answer[i] is the shortest distance from s[i] to the character c in s.

Example 1:

Input: s = "loveleetcode", c = "e"
Output: [3,2,1,0,1,0,0,1,2,2,1,0]

Example 2:

Input: s = "aaab", c = "b"
Output: [3,2,1,0]

Constraints:

• 1 <= s.length <= 104
• s[i] and c are lowercase English letters.
• c occurs at least once in s.

字符的最短距离。

给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。

我给出一个比较常规的思路。首先我们遍历一次input字符串，用hashmap记录每一个C的位置。第二遍遍历字符串的时候，对于每一个不是C的字母，我们比较一下当前这个字母的index和每一个字母C的index，找到其中的最小值。

时间O(n^2)

空间O(n)

Java实现

class Solution {
    public int[] shortestToChar(String s, char c) {
        int len = s.length();
        HashMap<Character, List<Integer>> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        // mark the index of all the Cs
        for (int i = 0; i < len; i++) {
            if (s.charAt(i) == c) {
                list.add(i);
            }
        }
        map.put(c, list);

        int[] res = new int[len];
        for (int i = 0; i < len; i++) {
            if (s.charAt(i) != c) {
                int min = Integer.MAX_VALUE;
                for (int index : map.get(c)) {
                    min = Math.min(min, Math.abs(index - i));
                }
                res[i] = min;
            } else if (s.charAt(i) == c) {
                res[i] = 0;
            }
        }
        return res;
    }
}

 

LeetCode 题目总结