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
相关推荐
梧桐树04291 小时前
python常用内建模块:collections
python
Dream_Snowar1 小时前
速通Python 第三节
开发语言·python
XH华1 小时前
初识C语言之二维数组(下)
c语言·算法
南宫生2 小时前
力扣-图论-17【算法学习day.67】
java·学习·算法·leetcode·图论
不想当程序猿_2 小时前
【蓝桥杯每日一题】求和——前缀和
算法·前缀和·蓝桥杯
落魄君子2 小时前
GA-BP分类-遗传算法(Genetic Algorithm)和反向传播算法(Backpropagation)
算法·分类·数据挖掘
菜鸡中的奋斗鸡→挣扎鸡2 小时前
滑动窗口 + 算法复习
数据结构·算法
蓝天星空2 小时前
Python调用open ai接口
人工智能·python
Lenyiin2 小时前
第146场双周赛:统计符合条件长度为3的子数组数目、统计异或值为给定值的路径数目、判断网格图能否被切割成块、唯一中间众数子序列 Ⅰ
c++·算法·leetcode·周赛·lenyiin
jasmine s3 小时前
Pandas
开发语言·python