4.10 百度笔试记录

百度太奇怪了,我3月7号投了简历,第一次笔试我有别的事放弃了,然后就一直没有再通知我笔试了,上周给HR发了一下邮件才赶上最后一次笔试。开个帖子记录一下今晚的编程题吧。

第一题太简单了就不po了,第二题大概是:有n个士兵,nums[i]代表第i个士兵的战斗值,小明和小红给士兵排序,两人共进行m次操作,小明每次操作会选择一个k,将前k个士兵从小到大排序;小红每次操作选择一个k,将前k个士兵从大到小排序,输出全部操作结束后士兵的战斗值。m和n的数量级在10^5。

简述一下思路,如果暴力的方法的话,复杂度是O(mnlogn),过不了;一个优化的方法是,一个操作的k如果大于等于它之前的操作的k,那之前的操作可以忽略,所以一个比较显然的思路是用单调栈来优化,用代码来解释吧,复杂度是O(nlogn + m)

n, m = map(int, input().split())
nums = list(map(int, input().split())) # 战斗值
stack, ans = [], []
for _ in range(m):
    p, k = map(int, input().split()) # p=1:小明操作;p=2:小红
    while stack and stack[-1][-1] <= k:
        stack.pop()
    stack.append((p, k))
stack.append((0, 0))
maxk, l, r = stack[0][1], 0, -1
lst = sorted(nums[:maxk])
for i in range(len(stack) - 1):
    length = stack[i][1] - stack[i + 1][1]
    if stack[i][0] == 1:
        ans.extend(lst[r:r - length:-1])
        r -= length
    elif stack[i][0] == 2:
        ans.extend(lst[l:l + length])
        l += length
ans = ans[::-1] + nums[maxk:]
print(*ans)

#暑期实习##我的实习求职记录#
全部评论
不戳,学到了单调栈,谢谢
点赞 回复 分享
发布于 2023-04-10 23:40 北京
我也是这个思路,if判断的,这为啥会超时
点赞 回复 分享
发布于 2023-04-11 00:13 北京
给力,好思路
点赞 回复 分享
发布于 2023-04-11 09:45 广东
cv吗
点赞 回复 分享
发布于 2023-04-13 21:44 上海

相关推荐

在拧螺丝的代码渣渣很喜欢后仰跳投:卖课的,拉黑就行了
点赞 评论 收藏
分享
评论
4
13
分享
牛客网
牛客企业服务