Optimizing Performance with HashSets in Python

Introduction

Greetings, programming enthusiast! In this unit, we're embarking on a thrilling numerical quest, where unidentified bridges connect the islands of data. On these bridges, we'll see hashes and bins, all converging into sets! Throughout our journey, we'll utilize the fundamental concepts of Python's built-in data type, the set, to formulate an optimal solution. So, fasten your seatbelt and get ready to solve problems!

Task Statement

The task for this unit is to devise a Python function that accepts two lists containing unique integers and returns another list containing the elements common to both input lists. This task provides an intriguing perspective on deriving similarities between two data sequences, a scenario commonly encountered in data comparisons and analytics.

For illustration, suppose we're given two lists:

Python
list1 = [1, 2, 3, 5, 8, 13, 21, 34]
list2 = [2, 3, 5, 7, 13, 21, 31]

The common_elements(list1, list2) function should comb through these arrays of integers and extract the common elements between them.

The expected outcome in this case should be:

Python
[2, 3, 5, 13, 21]

Brute Force Solution and Complexity Analysis

Before we delve into the optimized solution, it is instrumental to consider a basic or naïve approach to this problem and analyze its complexity. Often, our first intuitive approach is to iterate over both arrays in nested loops and find common elements. This way, for each element in the first list, we check for its presence in the second list. If it's found, it is added to our result set. Let's see how such a solution would look like:

Python
def common_elements_slow(list1, list2):
    common = [] # list to store common elements
    for num1 in list1:
        for num2 in list2:
            if num1 == num2:
                common.append(num1)
                break  # break inner loop as we found the number in list2
    return common

However, the problem with this approach lies in its efficiency. Given that the worst-case scenario has us traversing through every element in both lists, we refer to this as an O(n∗m)O(n*m) solution, where n and m represent the number of elements in list1 and list2 respectively. For large lists, this iterative approach tends to be inefficient and slow, making it a less desirable solution for this problem.

The solution we aim to implement in the following section utilizes a set data structure to optimize our algorithm and reach a solution in a markedly less computational time.

Sign up

Join the 1M+ learners on CodeSignal

Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal