Understanding K Element Patterns in LeetCode: The Basics (Part 1)
Shavon Harris
Posted on November 25, 2024
I finally understand! The best way to learn LeetCode is not by grinding through problem after problem, sometimes taking a whole hour to solve them inefficiently. The key to mastering LeetCode is studying patterns. Let's study a common one!
Interviewers love asking about finding, maintaining, or manipulating K elements in a string or array. At first, I thought each problem was totally different, but then I started seeing the connections. Let me show you what I mean with two problems that really helped me understand this pattern.
Problem 1: Find Subsequence of Length K with the Largest Sum
This is technically an 'easy' level question (only asked by one company), but, it teaches you so much about how to think about these k-element problems!
What They're Asking
You get an array of numbers and a value k, and you need to find k numbers from the array that add up to the biggest possible sum. BUT (and this is the part that tripped me up at first), you have to keep the numbers in their original order!
Example:
Oh! this one is trickier:
Ok I Get It
At first, I thought "just grab the k biggest numbers, done!" But nope - that order requirement changes everything. Here's what finally clicked:
- We need to remember where each number came from, right? So I thought, "I should pair every number with it's position?"
- Then we can sort these pairs by the values (that's the first number in each pair), but we're keeping track of where they came from (that's the second number)!
- Now the cool part - we only need k of these, so grab the first k pairs and just keep their positions:
- Finally, walk through the original array and only keep numbers whose positions are in our Set:
Here's The Code
2: Kth Largest Element in a Stream
Ok, so this one's also labeled 'easy' (asked by five companies), but this one was more confusing to me than any of the more difficult Kth element problems.
What They're Asking
Imagine you're working at a university and students keep submitting test scores. Your job is to always know the kth highest score at any time. New scores keep coming in, and you need to keep track.
They give you k and some initial scores, then they keep throwing new scores at you and want to know the kth highest each time. Let's look at an example:
Ok, I Think I Get It
At first, I kept trying to sort the entire array every time a new score came in but I know that sorting can be inefficient. Then I thought why am I keeping track of all the scores when I only care about the top k?
Here's how I broke it down:
- First, sort the initial scores and only keep the top k:
- Now when a new score comes in:
If it's smaller than our kth highest (first number), ignore it
If it's bigger, it belongs in our list somewhere
Here's what's happening with each add:
The Code
Why These Two Problems Are Connected
Both problems taught me something super important about handling k elements:
- First problem: Sometimes you need to track where elements came from
- Second problem: Sometimes you only need to keep k elements around
These k-element problems are all about being clever with what information you keep and what you throw away.
Next time we'll look at two more k-element problems that build on these idea. I hope at the end you see a pattern and these types of problems seem less scary!
Posted on November 25, 2024
Join Our Newsletter. No Spam, Only the good stuff.
Sign up to receive the latest update from our blog.