Solution: Largest Number After Digit Swaps by Parity
Let's solve the Largest Number After Digit Swaps by Parity using the Two Heaps pattern.
We'll cover the following
Statement
You are given a positive integer num
. You can swap any two digits of num
as long as they share the same parity (both are odd or both are even).
Your task is to return the largest possible value of num
after performing any number of such swaps.
Constraints:
num
Solution
The algorithm rearranges the digits of a given integer num
to form the largest possible number by allowing swaps only between digits that share the same parity (either odd or even). To achieve this, it uses two max-heaps—one for odd digits and one for even digits—which enable efficient storage and retrieval of the largest available digits in each parity group. As it iterates through the digits, the algorithm replaces each digit with the largest possible digit of the same parity from its respective heap. There’s no need to compare the current digit with the digit from the heap because the largest digit of each parity is always at the top of its heap, it can be popped directly and appended to the result. This ensures the final number is maximized while following to the parity constraint, ultimately producing the largest possible value that can be constructed from num
.
The steps of the algorithm are as follows:
Convert the integer
num
into a listdigits
that contains each individual digit.Create two empty max-heaps,
odd_heap
andeven_heap
, to store the odd and even digits, respectively.Iterate through each digit in the
digits
list:If the digit is even, push it onto
even_heap
.If the digit is odd, push it onto
odd_heap
.
Initialize an empty list
result
to construct the final number.Iterate through each digit in the
digits
list:If the digit is even, pop the digit from
even_heap
and append it toresult
.If the digit is odd, pop the digit from
odd_heap
and append it toresult
.
Join the elements of
result
into a single string, then convert it back to an integer and returnresult
.
Let’s look at the following illustration to get a better understanding of the solution:
Level up your interview prep. Join Educative to access 70+ hands-on prep courses.