Skip to main content

Two Sum Problem Explained in Python | HashMap Intuition & Optimal Approach

Two Sum Problem – From Confusion to Clear Intuition

The Two Sum problem is one of those classic interview questions that looks simple, but can easily trap you into an inefficient approach if you don’t recognize the pattern early. Initially, I also thought about checking every possible pair, but that quickly leads to a slow solution.

Problem Statement (In Simple Words)

You are given an array of integers and a target value. Your task is to return the indices of the two numbers such that they add up to the target. Each input has exactly one solution, and you cannot use the same element twice.

Brute Force Thought Process

The first idea that naturally comes to mind is: for every number, check all other numbers to see if their sum equals the target. While this works logically, it results in O(n²) time complexity, which is not optimal.

Key Insight That Changes Everything

Instead of asking: “Which two numbers add up to the target?”
I reframed the question as: “For the current number, have I already seen the number needed to complete the target?”

This small shift in thinking is what introduces the HashMap pattern.

Why HashMap Works Here

  • We store numbers we have already seen along with their indices
  • For each current number, we compute target - current number
  • If that value already exists in the map, we have found the answer

This allows us to solve the problem in a single pass.

Optimal Python Solution

Here is the clean and efficient solution using a HashMap:

class Solution:
    def twoSum(self, nums, target):
        t_sum = {}
        for index, i in enumerate(nums):
            if target - i in t_sum:
                return [t_sum[target - i], index]
            t_sum[i] = index

Complete Code on GitHub

You can find the complete implementation of this solution in my GitHub repository here:

🔗 Two Sum – Python Solution (GitHub)

Time & Space Complexity

  • Time Complexity: O(n) — single traversal of the array
  • Space Complexity: O(n) — extra space for the HashMap

Final Takeaway

The Two Sum problem is not just about solving one question— it teaches an important pattern: using extra space to reduce time complexity.

Once this intuition clicks, many array and HashMap problems become much easier to approach, especially in coding interviews and competitive programming.

Comments

Popular posts from this blog

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...

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...

Product of Array Except Self in Python | Prefix & Suffix Explained (LeetCode 238)

Problem Overview The Product of Array Except Self is a classic problem that tests your understanding of array traversal and optimization. The task is simple to state but tricky to implement efficiently. Given an integer array nums , you need to return an array such that each element at index i is equal to the product of all the elements in nums except nums[i] . The challenge is that: Division is not allowed The solution must run in O(n) time Initial Thoughts At first glance, it feels natural to compute the total product of the array and divide it by the current element. However, this approach fails because division is forbidden and handling zeroes becomes messy. This pushed me to think differently — instead of excluding the current element, why not multiply everything around it? That’s where the prefix and suffix product pattern comes in. Key Insight: Prefix & Suffix Products For every index i : Prefix product → product of all elements to t...