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