[LeetCode] 每日一题 Day 2：821. 字符的最短距离

题目

给定一个字符串 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 中的所有字母均为小写字母。

解决

思路:

弄个双指针，一个指向前一个匹配的下标，一个指向后一个匹配的下标，总共只需要遍历两遍,最后取两次遍历的最小值。

class Solution {
    public int[] shortestToChar(String s, char c) {
        int pre=-20000;
        int[] a=new int[s.length()];
        for(int i=0;i<s.length();i++){
            char ch=s.charAt(i);
            if(ch==c){
                pre=i;
            }
            a[i]=i-pre;
        }
        int next=20000;
        for(int i=s.length()-1;i>=0;i--){
            char ch1=s.charAt(i);
            if(ch1==c){
                next=i;
            }
            a[i]=Math.min(a[i],next-i);
        }
        return a;
    }
}

复杂度

时间复杂度 O(N)
空间复杂度 O(N)