【力扣每日一题】821. 字符的最短距离

题目描述

给你一个字符串 s 和一个字符 c ，且 c 是 s 中出现过的字符。

返回一个整数数组 answer ，其中 answer.length == s.length 且 answer[i] 是 s 中从下标 i 到离它 最近 的字符 c 的 距离 。

两个下标 i 和 j 之间的 距离 为 abs(i - j) ，其中 abs 是绝对值函数。

示例 1：

输入：s = "loveleetcode", c = "e"
输出：[3,2,1,0,1,0,0,1,2,2,1,0]
解释：字符 'e' 出现在下标 3、5、6 和 11 处（下标从 0 开始计数）。
距下标 0 最近的 'e' 出现在下标 3 ，所以距离为 abs(0 - 3) = 3 。
距下标 1 最近的 'e' 出现在下标 3 ，所以距离为 abs(1 - 3) = 2 。
对于下标 4 ，出现在下标 3 和下标 5 处的 'e' 都离它最近，但距离是一样的 abs(4 - 3) == abs(4 - 5) = 1 。
距下标 8 最近的 'e' 出现在下标 6 ，所以距离为 abs(8 - 6) = 2 。

示例 2：

输入：s = "aaab", c = "b"
输出：[3,2,1,0]

提示：

• 1 <= s.length <= 10^4
• s[i] 和 c 均为小写英文字母
• 题目数据保证 c 在 s 中至少出现一次

---

这里我采用了两种方法：

方法一：

额外声明一个数组cpos，存储s中c的索引，然后s中每个字符索引与cpos中的索引依次作差，取最小差值。

方法二：

对s中每一个字符，声明两个指针j,k，分别向左和向右移动，如果出现某一个位置i-j或者i+k，是字符c，则将j或k的位置写入答案数组中 。

class Solution {
    public int[] shortestToChar(String s, char c) {
        //方法1，慢
        // char[] cs = s.toCharArray();
        // int[] ans = new int[cs.length];
        // List<Integer> cpos = new ArrayList<>();
        // for(int i=0;i<cs.length;i++)
        //     if (cs[i]==c)
        //         cpos.add(i);
        // for(int i=0;i<cs.length;i++){
        //     int minDis = cs.length;
        //     for(int pos:cpos){
        //         int curDis = Math.abs(pos-i);
        //         minDis = curDis<minDis?curDis:minDis;
        //     }
        //     ans[i] = minDis;
        // }
        // return ans;

        //方法2，最快
        // char[] cs = s.toCharArray();
        // int[] ans = new int[cs.length];
        // for(int i=0;i<cs.length;i++){
        //     if(cs[i]==c){
        //         ans[i]=0;
        //         continue;
        //     }
        //     int j=0,k=0;
        //     while(true){
        //         if(i-(j+1)>=0) j++;
        //         if(i+(k+1)<=cs.length-1) k++;
        //         if(cs[i-j]==c){
        //             ans[i]=j;
        //             break;
        //         }
        //         if(cs[i+k]==c){
        //             ans[i]=k;
        //             break;
        //         }
        //     }
        // }
        // return ans;

        //方法3，较快
        // int[] ans = new int[s.length()];
        // for(int i=0;i<s.length();i++){
        //     if(s.charAt(i)==c){
        //         ans[i]=0;
        //         continue;
        //     }
        //     int j=0,k=0;
        //     while(true){
        //         if(i-(j+1)>=0) j++;
        //         if(i+(k+1)<=s.length()-1) k++;
        //         if(s.charAt(i-j)==c){
        //             ans[i]=j;
        //             break;
        //         }
        //         if(s.charAt(i+k)==c){
        //             ans[i]=k;
        //             break;
        //         }
        //     }
        // }
        // return ans;

    }
}

---

官方用了类似的思路，只不过是先求左边的最小距离，再求右边的最小距离，最后比较大小。

官方

class Solution {
    public int[] shortestToChar(String s, char c) {
        int n = s.length();
        int[] ans = new int[n];

        for (int i = 0, idx = -n; i < n; ++i) {
            if (s.charAt(i) == c) {
                idx = i;
            }
            ans[i] = i - idx;
        }

        for (int i = n - 1, idx = 2 * n; i >= 0; --i) {
            if (s.charAt(i) == c) {
                idx = i;
            }
            ans[i] = Math.min(ans[i], idx - i);
        }
        return ans;
    }
}

---

简单题，但是思路多样化，对于训练思维仍有不小帮助。