Skip to main content

Trapping Rain Water Explained – From Brute Force to Two Pointer Approach (LeetCode)

Trapping Rain Water – My Learning Journey (From Intuition to Two Pointers)

Today I worked on the Trapping Rain Water problem, and honestly, it was not easy at first. I couldn’t directly come up with the solution, so I decided to step back and focus on understanding the intuition instead of forcing the code.

To build that intuition, I watched this video, which helped me understand how water is actually trapped between bars:

Reference Video:
https://www.youtube.com/watch?v=TCaBnVIllrQ


Key Intuition

The most important realization for this problem is:

Water trapped at any index depends on the maximum height on its left and the maximum height on its right.

Mathematically:

water[i] = min(max_left, max_right) - height[i]

If this value is negative, no water can be trapped at that position.


Approach 1: Brute Force (Time Limit Exceeded)

My first approach was very straightforward. For each index, I calculated:

  • Maximum height on the left
  • Maximum height on the right

However, I used slicing inside a loop, which created subarrays repeatedly and caused a Time Limit Exceeded (TLE) error.


class Solution:
    def trap(self, height):
        rm = []
        lm = []
        for i in range(len(height)):
            rm.append(max(height[:i+1]))
            lm.append(max(height[i:]))

        result = 0
        for i in range(len(height)):
            result += min(rm[i], lm[i]) - height[i]

        return result

This helped me understand the logic, but it was inefficient.


Approach 2: Prefix Maximum Arrays (Optimized)

To fix the performance issue, I optimized the solution by precomputing:

  • left_max array
  • right_max array

This reduced the time complexity to O(n) while using extra space.


class Solution:
    def trap(self, height):
        n = len(height)
        lm = [0] * n
        rm = [0] * n

        lm[0] = height[0]
        for i in range(1, n):
            lm[i] = max(lm[i-1], height[i])

        rm[n-1] = height[n-1]
        for i in range(n-2, -1, -1):
            rm[i] = max(rm[i+1], height[i])

        result = 0
        for i in range(n):
            result += min(lm[i], rm[i]) - height[i]

        return result

At this point, the logic was clear, but I still wanted to understand how this could be solved using two pointers.


Approach 3: Two Pointer Technique (Optimal)

The key insight for the two-pointer approach is:

Water is always limited by the smaller boundary.

Instead of storing arrays, I maintained:

  • left_max while moving from the left
  • right_max while moving from the right

At each step, I moved the pointer with the smaller maximum height, because that side determines how much water can be trapped.


class Solution:
    def trap(self, height):
        l, r = 0, len(height) - 1
        l_max = r_max = 0
        result = 0

        while l < r:
            l_max = max(l_max, height[l])
            r_max = max(r_max, height[r])

            if l_max < r_max:
                result += l_max - height[l]
                l += 1
            else:
                result += r_max - height[r]
                r -= 1

        return result

This solution runs in O(n) time and uses O(1) extra space. More importantly, I finally understood why it works, instead of just memorizing it.


Final Thoughts

Initially, I couldn’t build the solution on my own, but instead of giving up, I focused on understanding the intuition and building the solution step by step.

This problem taught me an important lesson:

Understanding the pattern matters more than rushing to the final code.

Now, the two-pointer approach feels natural, and I feel more confident approaching similar problems in the future.

Comments

Popular posts from this blog

How do I run Python on Google Colab using android phone?

Regardless of whether you are an understudy keen on investigating Machine Learning yet battling to direct reproductions on huge datasets, or a specialist playing with ML frantic for extra computational force, Google Colab is the ideal answer for you. Google Colab or "the Colaboratory" is a free cloud administration facilitated by Google to support Machine Learning and Artificial Intelligence research, where frequently the obstruction to learning and achievement is the necessity of gigantic computational force. Table of content- What is google colab? how to use python in google colab? Program to add two strings given by the user. save the file in google colab? What is google colab? You will rapidly learn and utilize Google Colab on the off chance that you know and have utilized Jupyter notebook previously. Colab is fundamentally a free Jupyter notebook climate running completely in the cloud. In particular, Colab doesn't need an arrangement, in addition to the notebook tha...

Introducing CodeMad: Your Ultimate Universal IDE with Custom Shortcuts

Introducing CodeMad: Your Ultimate Multi-Language IDE with Custom Shortcuts Welcome to the world of CodeMad, your all-in-one Integrated Development Environment (IDE) that simplifies coding and boosts productivity. Developed in Python, CodeMad is designed to make your coding experience smoother and more efficient across a variety of programming languages, including C, C++, Java, Python, and HTML. Whether you're a beginner or an experienced programmer, CodeMad is your go-to tool. In this blog, we'll dive deep into the workings of CodeMad, highlighting its unique features and easy installation process. The Power of Shortcuts CodeMad's intuitive interface is built around a set of powerful keyboard shortcuts that make coding a breeze. Here are some of the key shortcuts you'll find in CodeMad: Copy (Ctrl+C) : Duplicate text with ease. Paste (Ctrl+V) : Quickly insert copied content into your code. Undo (Ctrl+Z) and Redo (Ctrl+Y) : Correct mistakes and s...

LeetCode 88 Explained: Four Approaches, Mistakes, Fixes & the Final Optimal Python Solution

Evolving My Solution to “Merge Sorted Array” A practical, beginner-friendly walkthrough showing four versions of my code (from a naive approach to the optimal in-place two-pointer solution). Includes explanations, complexity and ready-to-paste code. Problem Summary You are given two sorted arrays: nums1 with size m + n (first m are valid) nums2 with size n Goal: Merge nums2 into nums1 in sorted order in-place . Version 1 — Beginner Approach (Extra List) I merged into a new list then copied back. Works, but not in-place and uses extra memory. class Solution: def merge(self, nums1, m, nums2, n): result = [] p1 = 0 p2 = 0 for _ in range(m+n): if p1 >= m: result.extend(nums2[p2:n]) break elif p2 >= n: result.extend(nums1[p1:m]) break elif nu...