LeetCode.2940.找到Alice和Bob可以相遇的建筑

友情提示:这个方法并没有通过案例,只通过了944个案例(很难受),超时了,但是想着还是分享出来吧

题目描述:

给你一个下标从 0 开始的正整数数组 heights ,其中 heights[i] 表示第 i 栋建筑的高度。

如果一个人在建筑 i ,且存在 i < j 的建筑 j 满足 heights[i] < heights[j] ,那么这个人可以移动到建筑 j 。

给你另外一个数组 queries ,其中 queries[i] = [ai, bi] 。第 i 个查询中,Alice 在建筑 ai ,Bob 在建筑 bi 。

请你能返回一个数组 ans ,其中 ans[i] 是第 i 个查询中,Alice 和 Bob 可以相遇的 最左边的建筑 。如果对于查询 i ,Alice和Bob 不能相遇,令 ans[i] 为 -1

输入输出实例:

思路:我的思路是先定义一个长度为len(queries)的全为-1的列表ans,然后遍历queries里的每组元素,然后找alice和bob哪个在右边,然后从在右边的那个下标为起始开始遍历heights列表直到遇到一个满足条件的建组就结束这次循环,开始找下一组。

那么对于在找每组的我们要的下标过程为:如果这两个在同一个建筑,那我们直接将这个建筑(也就是下标)放到ans列表中;如果是bob在右边的建筑,我们就要找满足条件的建筑即:heightsj >= heightsbob and heightsj > heightsalice,这个条件把他们汇合的建筑是bob所在建筑也考虑到;同理可得alice在右边我们如何做。根据上述思路有以下代码:

python 复制代码
class Solution:
    def leftmostBuildingQueries(self, heights: List[int], queries: List[List[int]]) -> List[int]:
        ans = [-1]*len(queries)
        for i in range(len(queries)):
            alice,bob = queries[i][0],queries[i][1]
            #a对应的是heights[alice],b是heights[bob]
            if alice == bob :
                ans[i] = bob
                continue
            if alice < bob:
                for j in range(bob,len(heights)):
                    if heights[j] >= heights[bob] and heights[j] > heights[alice]:
                        ans[i] = j
                        break
            else:
                for j in range(alice,len(heights)):
                    if heights[j] >= heights[alice] and heights[j] > heights[bob]:
                        ans[i] = j
                        break
        return ans
相关推荐
Ivanqhz4 分钟前
层归一化、残差、前馈网络与激活函数简述
服务器·数据库·人工智能·深度学习·算法
BD_Marathon6 分钟前
消息对象中字段的说明
java·前端·python
Java后端的Ai之路12 分钟前
Python进阶探索23 - Python中的Time与Datetime模块
开发语言·数据库·python·datetime·time
不会就选b1 小时前
算法日常・每日刷题--<动态规划>3
数据结构·算法
Java后端的Ai之路1 小时前
Python进阶探索24_应用案例_linux系统的监控
linux·开发语言·python·应用·探索
happylifetree1 小时前
Python22-26:核心语法-流程控制语句-if条件判断
python
程序员清风1 小时前
PydanticAI 实战:用类型安全构建可靠的 Python Agent
开发语言·python·安全
-dzk-2 小时前
【堆】LC 347.前 K 个高频元素
算法·堆
H.莓飛2 小时前
【数据结构】栈_OJ题
linux·数据结构·算法·centos
梦想画家2 小时前
SQLMesh Python 模型入门(三):前后置语句、蓝图建模与避坑指南
大数据·python·sqlmesh