接雨水——单调栈（python）

news2026/4/9 7:41:44

思路利用栈的先进后出后进先出特性。使用单调栈入栈下标。当遇到更高的墙时说明形成了凹槽弹出栈元素开始计算接水量。每次弹出栈后记得要判空因为这里用的是大于要注意和求柱状图中最大矩形区别开。def rain(nums): #单调栈 stack[] #先进后出特性 res0 for i in range(len(nums)): while stack and nums[i]nums[stack[-1]]: #当遇到更高的墙说明形成了凹槽 tmpstack.pop() if not stack: #因为判断的是大于所以每弹出一次需要判断是否空 break leftstack[-1] #左边的墙 hnums[tmp] #当前高度 wi-left-1 #宽度 curHeightmin(nums[left],nums[i])-h #接水量取决于较小高度 resw*curHeight #接水量 stack.append(i) print(res) return res def main(): lineinput() numslist(map(int,line.split())) rain(nums) if __name____main__: main()

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2498731.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！