算法随笔_21:字符的最短距离

上一篇:算法随笔_20:区间子数组个数 -CSDN博客

=====================

题目描述如下:

给你一个字符串 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 。

=====================

算法思路:

对于下标i到离它最近的字符 c 的距离,我们设为dst_i。我们可以把dst_i它分开来看:

  1. 离下标i左侧最近的距离dstL_i。

  2. 离下标i右侧最近的距离dstR_i。

我们取两个距离最近的那个值即是dst_i。

对于计算dstL_i,我们先初始一个变量c_indL=-1。然后从左往右遍历数组,每访问一个元素,我们判断这个元素是否是字符c。

如果是的话,我们把它的下标记录到变量c_indL中,这样的话变量c_indL永远是离正在访问元素最近的那个。

如果不是字符c,且c_indL大于等于0,那么我们可以记录下dstL_i就等于当前的元素下标减去c_indL。如果c_indL为-1,那就说明当前元素的左侧没有字符c,它的距离应该是无限大。算法实现中我们可以设置当前元素的dstL_i为字符串的长度s.length。因为对于dst_i,它的最大可能值只能是s.length-1。

这样,一次遍历下来,所有元素的dstL_i都已经找到,我们把这些数值存储到数组dstL中。

同理,我们可以从右往左遍历数组,找出所有元素的dstR_i的值,我们把这些数值存储到数组dstR中。

最后再遍历一遍整个数组,取两个距离dstL_i和dstR_i的最小值做为dst_i,并存储到dst数组中。

此算法的时间复杂度为O(n)

下面是代码实现:

复制代码
class Solution(object):
    def shortestToChar(self, s, c):
        """
        :type s: str
        :type c: str
        :rtype: List[int]
        """
        s_len=len(s)
        dstL=[0]*s_len
        dstR=[0]*s_len
        dst=[]
        c_indL=-1
        c_indR=-1
        for i in range(s_len):
            tmp_cL=s[i]
            if tmp_cL==c:
                c_indL=i
                dstL[i]=0
            elif c_indL==-1:
                dstL[i]=s_len
            else:
                dstL[i]=i-c_indL
                
            r_ind=s_len-i-1
            tmp_cR=s[r_ind]
            if tmp_cR==c:
                c_indR=r_ind
                dstR[r_ind]=0
            elif c_indR==-1:
                dstR[r_ind]=s_len
            else:
                dstR[r_ind]=c_indR-r_ind
                
        for i in range(s_len):
            dst.append(min(dstL[i],dstR[i]))
        return dst     
相关推荐
ZhengEnCi7 小时前
Agent 架构怎么选:从单 Agent 到多 Agent 编排的选型决策指南
算法·架构
秋天的怀念_David7 小时前
第二十二夜 · 数字迷雾
java·程序人生·算法
Pniubi8 小时前
力扣283移动零(双指针法)
算法·leetcode·职场和发展
hhb_6188 小时前
大模型落地三重优化实战
人工智能·算法·机器学习
云和数据.ChenGuang8 小时前
光哥说Python RAG 系统需要哪些评估指标:从检索到生成的全链路评测指南
网络·人工智能·python·算法·langchain
2601_970703839 小时前
工业除湿机租赁:把一份租金合同拆开看
大数据·数据结构·算法·数据库架构·推荐算法
心中有你02149 小时前
Java实现Dijkstra最短路径算法(邻接表 + 优先队列)
java·数据库·算法
ZhengEnCi11 小时前
同样的模型,为什么本地和云 API 差别那么大:量化精度、聊天模板与采样参数的三大元凶
人工智能·算法·llm
Flynn Liang11 小时前
RAG 答非所问/幻觉怎么破?Spring AI QuestionAnswerAdvisor 召回优化实战(TopK/重排/混合检索)
人工智能·算法·spring·rag·幻觉·springai
liminla!11 小时前
SQL 从入门到进阶:语法速查与实战练习(完整优化版)
数据库·sql·算法