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

ALLe_Y2024-08-11 6:04

友情提示：这个方法并没有通过案例，只通过了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