解锁SQL无限可能 | 如何利用SQL求解力扣难题接雨水问题？

目录

1  需求描述

1 数据准备

2 问题分析

3 小结

---

1  需求描述

原文链接：42. 接雨水 - 力扣（LeetCode）

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

输入：height = [4,2,0,3,2,5]
输出：9

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

1 数据准备

with rain_data as
(
  select array(0,1,0,2,1,0,1,3,2,1,2,1) height
)
select * from rain_data;

 

2 问题分析

本文为力扣第42题，属于hard级别。分析该问题的关键点在于"找规律，抓特征"，这也是我专栏里一直提的核心技巧，语言不重要，重要的是思维方法，是算法逻辑，其余的工作只是利用语言去翻译，这一点对SQL语言尤为重要，因为语法简单的语言在实现上一定要重思维，重逻辑。

 这道题其实就是“木桶原理”，简单的说就是一个柱子能接多少水，取决于它两边“较短的板”。另外一个前提条件就是，两边的柱子高度都要比所要装水的柱子的高度要高，否则肯定是无法装水的。因此我们只需要关注左边最高的木板和右边最高的模板中较矮的一个就够了，那么存储的水，等于两边木板的较小值减去当前高度的值。用公式表示如下：

res[i] = min(l_max[i], r_max[i]) - height[i];

 我们将上述公式翻译成SQL语言即可：

第一步：先将数组展开成行

select tmp.pos + 1 id
          , tmp.height  height
     from rain_data
              lateral view posexplode(height) tmp as pos, height

 第二步：求当前位置处，左最大与右最大

select  id
    , height
    , max(height) over(order by id) l_max
    , max(height) over(order by id desc) r_max
from
    (select tmp.pos + 1 id
          , tmp.height  height
     from rain_data
              lateral view posexplode(height) tmp as pos, height
     )
        t;

 第三步：利用公式标记接雨水的柱子

select id
      ,height
      ,least(l_max,r_max) - height     as rain_flg
from
(select id
      , height
      , max(height) over (order by id)      l_max
      , max(height) over (order by id desc) r_max
 from (select tmp.pos + 1 id
            , tmp.height  height
       from rain_data
                lateral view posexplode(height) tmp as pos, height) t) t
order by id

 第四步：求出最终结果值。完整的SQL如下：

select sum(rain_flg)
from
(select least(max(tmp.height) over (order by id), max(tmp.height) over (order by id desc)) - tmp.height rain_flg
 from rain_data r
          lateral view posexplode(r.height) tmp as id, height
 ) t

3 小结

本题给出上述结果，在SQL层面已经算完成了，SQL查询主要的任务就是关注特征，找出问题的特征，利用SQL提取出来就可以了，其优化方式和代码语言有所不同，对于写代码的形式，上述属于暴力求解，需要优化其两边搜所的效率，因此就会有双指针和动态规划的求解思路。虽然，但是，但对于本文一开始而言找出”木桶原理”的特征（res[i] = min(l_max[i], r_max[i]) - height[i]），也不是一件容易得事情，需要一定问题的分析能力。

关于此类如何找规律、抓特征这一思维技巧，我把他放在我的专栏数字化建设通关指南专栏里，并把所有的规律技巧整理在如下文章中，链接如下：

SQL很简单，可你却写不好？也许这才是SQL最好的教程-CSDN博客

这篇文章我将持续更新，欢迎关注并收藏，也欢迎向我提问题，我的公众号“会飞的一十六”也开通了，里面的内容和评论区也很精彩