LeetCode: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 中的所有字母均为小写字母。

题解

题解1

记录字符串S中字符C出现的位置,判断字符串中每一个元素与字符C的距离,求其最小值。

class Solution {
public:
    vector<int> shortestToChar(string S, char C) {
        int n = S.size();
        vector<int>location_c,res;
        for(int i=0;i<n;i++){
            if(C==S[i]) location_c.push_back(i);
        }
        for(int i=0;i<n;i++){
            int dis = n;//求最小距离时一般赋值为相对应的最大值,比如此时数组元素最大距离为数组长度.
            for(int j=0;j<location_c.size();j++){
                int distance = abs(location_c[j]-i);
                dis = min(dis,distance);
            }
            res.push_back(dis);
        }
    return res;
    }
};

复杂度分析

• 时间复杂度: O(nk), n为字符串的长度,k为字符串s中字符C的数量。
• 空间复杂度: O(n)。

题解2

从左向右遍历, 记录当前字符S[i]左边最近一个字符C的位置,如果字符C暂时未出现,暂定距离为无穷大(或数组长度)。
从右向左遍历, 记录当前字符S[i]右边最近一个字符C的位置。

class Solution {
public:
   vector<int> shortestToChar(string S, char C) {
       vector<int> ans(S.size(), 0);
       //int prev = -10000;
       int prev = INT_MIN / 2;
       for(int i = 0; i < S.size(); i ++){
           if(S[i] == C) prev = i;
           ans[i] = i - prev;
       }
       //prev = 20000;
       prev = INT_MAX / 2;
       for(int i = S.size() - 1; i >= 0; i --){
           if(S[i] == C) prev = i;
           ans[i] = min(ans[i], prev - i);
       }
       return ans;
   }
};

注:

1. 为什么INT_MIN要除以2：
   因为int占4字节32位，根据二进制编码的规则，INT_MAX = 2³¹-1，INT_MIN= -2³¹，所有超过该限值的数，都会出现溢出。如果不除以2,在开始时,i-prev=i+2³¹会溢出！
2. INT_MAX不会产生溢出，INT_MAX可以不除以2。

复杂度分析

时间复杂度：O(n)，其中n是 S 的长度，我们需要遍历字符串两次。
空间复杂度：O(n)，ans 数组的大小。

参考

字符的最短距离（方法一+方法二+方法二优化）
【C++】两种思路
C++简洁代码（简单易懂且双百）：