LeetCode 42. Trapping Rain Water

Python解法

解题思路:

本思路需找到最高点左右遍历,时间复杂度O(nlogn),以下为向左遍历的过程。

  1. 将每一个点的高度和索引存成一个元组 (val, idx)
  2. 找到最高的点(可能有多个,任取一个),记为 (now_val, now_idx)。
  3. 向左找第一个val不大于now_val的点(left_val, left_idx)。
  4. 以left_val作为水平面,用height[left_val+1, now_val-1]中每一个点更新ans。
  5. now_val, now_idx = left_val, left_idx。
  6. 如果now_idx >= 0 ,重复执行3;否则执行7。
  7. 同理从最高点向右遍历更新ans。

向左找的具体操作

  1. 将每一个点的高度和索引存成一个元组 (val, idx)
  2. 将元组按val从大到小排序,若val相等,按idx从大到小排序。实现过程:直接sort,然后reverse。
  3. 从左向右遍历,找到第一个idx小于now_idx的节点(left_val, left_idx),用height[left+1, now-1]中的点更新ans。
  4. 更新now节点为left节点。
  5. 如果now_idx >= 0 ,重复执行3;否则结束循环,开始向右找。

向右找的具体操作

  1. 将每一个点的高度和索引存成一个元组 (val, idx)
  2. 将元组按val从大到小排序,若val相等,按idx从小到大排序。实现过程:用functools中的cmp_to_key重载排序过程。
  3. 从左向右遍历,找到第一个idx大于now_idx的节点(right_val, right_idx),用height[now+1, right-1]中的点更新ans。
  4. 更新now节点为lright节点。
  5. 如果now_idx >= 0 ,重复执行3;否则执行6。
  6. 输出ans。
注意:向左向右只有排序不同,其他操作类似,

利用functools中的cmp_to_key重载排序过程代码如下:

def mcmp(a, b):
val1,idx1 = a
val2,idx2 = b
if val1 == val2:
return idx2-idx1
return val1-val2 from functools import cmp_to_key
l = [] # l中存元组(val, idx)
l.sort(reverse=True, key=cmp_to_key(Solution.mcmp))

代码如下

class Solution:
# 利用functools中的cmp_to_key重载排序:
def mcmp(a, b):
val1,idx1 = a
val2,idx2 = b
if val1 == val2:
return idx2-idx1
return val1-val2 def trap(self, height: List[int]) -> int:
if len(height) == 0:
return 0
from functools import cmp_to_key
l = []
# 将(val, idx)存入l = []中
for i in range(0, len(height)):
l.append((height[i],i)) ans = 0 # 往左找
l.sort(reverse=True)
h, con = l[0]
for i in range(1, len(l)):
val, left = l[i]
if left < con:
for j in range(left+1, con):
ans += val - height[j]
con = left # 往右找
h, con = l[0]
l.sort(reverse=True, key=cmp_to_key(Solution.mcmp))
for i in range(1, len(l)):
val, right = l[i]
if right > con:
for j in range(con+1, right):
ans += val - height[j]
con = right
return ans

C++解法

解题思路:

本思路与Python类似,需找到最高点左右遍历,时间复杂度O(nlogn)以下为向左遍历的过程。差别在于C++是存map,无须手动排序。

// m中first存高度val,second存索引集合。
// set<int> s = m[2]表示高度为2的点的位置集合。
map<int, set<int>> m;

实现代码如下:

class Solution {
public:
int trap(vector<int>& height) {
if(height.size() == 0) return 0;
int ans = 0;
// m中first存高度val,second存索引集合。
// set<int> s = m[2]表示高度为2的点的位置集合。
map<int, set<int>> m;
// itm表示m的一个迭代器
map<int, set<int>>::iterator itm;
set<int> s;
set<int>::iterator its;
//将(val, idx)存入map
int sz = height.size();
for(int i = 0;i < sz; ++i){
m[height[i]].insert(i);
} //向左找
itm = --m.end();
s = itm->second;
int h = itm->first;
int left = *s.begin();
//每次找val小,idx小的
while(left != -1 && left >= 0){
s = m[h];
its = s.lower_bound(left);
if(its == s.begin()){
if(itm == m.begin()){
left = -1;
}else{
--itm;
h = itm->first;
}
}else{
--its;
for(int i = *its+1;i < left; ++i){
ans += h-height[i];
}
left = *its;
continue;
}
} //向右找
itm = --m.end();
s = itm->second;
h = itm->first;
int right = *s.begin();
//每次找val小,idx大的
while(right != -1 && right < sz){
s = m[h];
its = s.upper_bound(right);
if(its == s.end()){
if(itm == m.begin()){
right = -1;
}else{
--itm;
h = itm->first;
}
}else{
for(int i = right+1;i < *its; ++i){
ans += h-height[i];
}
right = *its;
continue;
}
}
return ans;
}
};

LeetCode 42. Trapping Rain Water 【两种解法】(python排序遍历,C++ STL map存索引,时间复杂度O(nlogn))的更多相关文章

  1. leetcode#42 Trapping rain water的五种解法详解

    leetcode#42 Trapping rain water 这道题十分有意思,可以用很多方法做出来,每种方法的思想都值得让人细细体会. 42. Trapping Rain WaterGiven n ...

  2. [array] leetcode - 42. Trapping Rain Water - Hard

    leetcode - 42. Trapping Rain Water - Hard descrition Given n non-negative integers representing an e ...

  3. LeetCode - 42. Trapping Rain Water

    42. Trapping Rain Water Problem's Link ------------------------------------------------------------- ...

  4. [LeetCode] 42. Trapping Rain Water 收集雨水

    Given n non-negative integers representing an elevation map where the width of each bar is 1, comput ...

  5. leetCode 42.Trapping Rain Water(凹槽的雨水) 解题思路和方法

    Trapping Rain Water Given n non-negative integers representing an elevation map where the width of e ...

  6. [leetcode]42. Trapping Rain Water雨水积水问题

    Given n non-negative integers representing an elevation map where the width of each bar is 1, comput ...

  7. [LeetCode] 42. Trapping Rain Water 解题思路

    Given n non-negative integers representing an elevation map where the width of each bar is 1, comput ...

  8. Java [Leetcode 42]Trapping Rain Water

    题目描述: Given n non-negative integers representing an elevation map where the width of each bar is 1, ...

  9. LeetCode 42 Trapping Rain Water(积水体积)

    题目链接: https://leetcode.com/problems/trapping-rain-water/?tab=Description   Problem: 根据所给数组的值,按照上图的示意 ...

随机推荐

  1. 用批处理实现垃圾文件清除/自动关机/清除copy病毒

    晚上睡觉之前为了下emule经常使用命令shutdown,最近受一个小程序影响想做个自动关机的批处理文件免的麻烦!网上有高手做了个,不过运行时出 现一个绑定错误,at也不能执行,所以后来自己做了简化版 ...

  2. javaee IO流打印一行的方式

    package Dayin; import java.io.BufferedReader; import java.io.FileNotFoundException; import java.io.F ...

  3. BZOJ 1106: [POI2007]立方体大作战tet 树状数组 + 贪心

    Description 一个叫做立方体大作战的游戏风靡整个Byteotia.这个游戏的规则是相当复杂的,所以我们只介绍他的简单规 则:给定玩家一个有2n个元素的栈,元素一个叠一个地放置.这些元素拥有n ...

  4. vue-cli index.js dev 配置中 assetsPublicPath 的值不能填 "./" 的问题

    问题 使用nginx又代理了一层 在浏览器中 / 代表域名的根目录,./代表当前路径 线上发布的时候一般都会使用nginx反向代理,所以使用./是最靠谱的,但是vue-cli dev 中的 asset ...

  5. POJ3253 Fence Repair【贪心】

    我们的小伙伴Bingo真的很调皮,他在上课的路上看到树上有个鸟窝,他就想去把他捅下来,但是鸟窝很高他够不到,于是他就到处找木棍,想把这些木棍接在一起,然后去捅鸟窝.他一共找了N跟木棍 (1 ≤ N ≤ ...

  6. 【数据结构】4.1图的创建及DFS深度遍历(不完善)

    声明:本代码仅供参考,根本就不是正确代码(至少在我看来,有很多BUG和不完美的地方) 图的存储方式选择为邻接表,并且headNode只是来存储一个链表的Node首地址额 总之这个代码写的很垃圾呀很垃圾 ...

  7. Centos 7.x 源码编译搭建Nginx

    环境: centos 7 防火墙关闭 Selinx关闭 Nginx Web安装 安装依赖库 yum install pcre-devel pcre gcc gcc-c++ zlib zlib-deve ...

  8. Linux启用ftp服务及连接

    虚拟机的系统是centos6.3 第一步.启动ftp service vsftpd restart 提示 vsftpd: 未被识别的服务 解决方法是升级vsftpd服务 yum install vsf ...

  9. 【ACM】hdu_zs2_1003_Problem C_201308031012

    Problem C Time Limit : 2000/1000ms (Java/Other)   Memory Limit : 65536/32768K (Java/Other)Total Subm ...

  10. C#中的stathread标签【待填的坑】

    stathread这种线程是给COM组件使用的线程,如果不适用com对象 如果com对象标记为sta的,则它就是单线程运行的 stathread 组件线程遗留的标签