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在右边的建筑,我们就要找满足条件的建筑即:heights[j] >= heights[bob] and heights[j] > heights[alice],这个条件把他们汇合的建筑是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
相关推荐
芜湖xin18 分钟前
【题解-洛谷】P1706 全排列问题
算法·dfs
蓝婷儿29 分钟前
6个月Python学习计划 Day 16 - 面向对象编程(OOP)基础
开发语言·python·学习
chao_7891 小时前
链表题解——两两交换链表中的节点【LeetCode】
数据结构·python·leetcode·链表
大霞上仙2 小时前
nonlocal 与global关键字
开发语言·python
曦月逸霜2 小时前
第34次CCF-CSP认证真题解析(目标300分做法)
数据结构·c++·算法
Mark_Aussie2 小时前
Flask-SQLAlchemy使用小结
python·flask
程序员阿龙2 小时前
【精选】计算机毕业设计Python Flask海口天气数据分析可视化系统 气象数据采集处理 天气趋势图表展示 数据可视化平台源码+论文+PPT+讲解
python·flask·课程设计·数据可视化系统·天气数据分析·海口气象数据·pandas 数据处理
ZHOU_WUYI2 小时前
Flask与Celery 项目应用(shared_task使用)
后端·python·flask
且慢.5893 小时前
Python_day47
python·深度学习·计算机视觉