2007. Find Original Array From Doubled Array


Problem Description

This problem provides an array called changed, which has been created by taking an original array of integers, doubling each element, appending these doubled numbers to the original array, and then shuffling the entire collection of numbers.

Our task is to reconstruct the original array from the changed array. However, there are some constraints:

  1. If the changed array has an odd number of elements, it's impossible to obtain the original array since doubling each element of the original array and then merging would result in an even number of total elements.
  2. For each element in the original array, there must be a corresponding element in changed that is double its value.
  3. If changed does not satisfy the property of being a "doubled array,” as described, we need to return an empty array.

We are allowed to return the elements of the original array in any order, adding flexibility to our solution.

Intuition

To find the solution, the following intuition can guide us:

  1. Sorting: We start by sorting changed since this will arrange all double elements after their corresponding originals (since 2 * x > x for all x > 0). Sorting also helps us to handle duplicates effectively.

  2. Counting: Using a counter (frequency map) over the sorted changed array helps in keeping track of the elements we've used for forming the original array and the elements that need to be paired with their doubles.

  3. Pairing and Elimination: We iterate over the sorted changed array and look for the double of each element. If for any element x, the element 2x does not exist in sufficient quantity (or does not exist at all), we cannot form the original array, hence, we return an empty array.

  4. Building the Original Array: If an element x and its double 2x are found, we pair them and reduce their count in the frequency map. The element x is added to the original array.

  5. Validation: In the end, if the generated original array has exactly half the length of the changed array, we have successfully reformed the original array. Otherwise, the changed array wasn't a doubled array, to begin with, or we couldn't pair all elements correctly, and we return an empty array.

Using this approach caters to all conditions provided in the problem, thus guaranteeing a correct solution when one is possible.

Learn more about Greedy and Sorting patterns.

Solution Approach

The implementation follows the intuition closely using Python's built-in data structures and sorting algorithm.

Here's the step-by-step breakdown of the solution approach:

  1. Check for the Odd Number of Elements: A quick check to confirm if the length of the changed array is odd. Since the doubled array should be even in length, return an empty array immediately if this is the case.

    n = len(changed)
    if n & 1:
        return []
  2. Use a Counter for Frequency Tracking: A Counter object from the collections module is utilized to keep a frequency map of the elements in changed.

    cnt = Counter(changed)
  3. Sort the changed Array: The changed array is sorted to ensure that the elements and their doubles are aligned in increasing order.

    changed.sort()
  4. Iterate to Build the original Array: A for-loop iterates over each element in the sorted changed array, applying the following logic:

    • Skip Processed Elements: If an element x has already been used (i.e., its count is 0), skip it.
      if cnt[x] == 0:
          continue
    • Check for Double's Existence: If there is not enough of the double of x remaining (i.e., cnt[2 * x] is 0 or negative), the changed cannot be a doubled array, so return an empty list.
      if cnt[x * 2] <= 0:
          return []
    • Add Element to original and Update Counts: If a valid double is found, append x to the original array and decrement the counts for x and 2 * x.
      ans.append(x)
      cnt[x] -= 1
      cnt[x * 2] -= 1
  5. Final Validation and Return: After the loop is done, check if the length of the original array is exactly half the changed array (which implies each element was paired correctly). If not, the changed array couldn't have been a doubled array, and thus, return an empty array.

    return ans if len(ans) == n // 2 else []

By utilizing a sorted array and a frequency map, the algorithm ensures that all elements can be paired with their corresponding doubles, maintaining linearithmic time complexity due to sorting, with the remainder of operations being linear within the sorted array. The space complexity is linear due to the extra space used for the Counter and the resulting original array.

Ready to land your dream job?

Unlock your dream job with a 2-minute evaluator for a personalized learning plan!

Start Evaluator

Example Walkthrough

Let's illustrate the solution approach with a small example. Suppose we have the following changed array:

changed = [1, 3, 4, 2, 6, 8]

Step-by-step, we perform the following actions:

  1. Check for Odd Number of Elements: The length of the changed array is 6, which is even. We can proceed.

  2. Use a Counter for Frequency Tracking: Generate a frequency map:

    cnt = Counter([1, 3, 4, 2, 6, 8]) # Counter({1: 1, 3: 1, 2: 1, 4: 1, 6: 1, 8: 1})
  3. Sort the changed Array: The sorted changed array looks like this:

    changed = [1, 2, 3, 4, 6, 8]
  4. Iterate to Build the original Array:

    • For the first element 1, we find that its double 2 exists, so we add 1 to original and update the counts:

      ans = [1]
      cnt = {1: 0, 3: 1, 2: 0, 4: 1, 6: 1, 8: 1}
    • We skip 2 since its count is now 0.

    • Next, for 3, we find its double 6, add 3 to original, and update counts:

      ans = [1, 3]
      cnt = {1: 0, 3: 0, 2: 0, 4: 1, 6: 0, 8: 1}
    • We skip 4 as we don't have 8 (double of 4) in enough quantity (count is 0), and since we can't find a valid double, we would normally return an empty array. However, for demonstration purposes, we will assume the double 8 exists and continue:

      ans = [1, 3, 4]
      cnt = {1: 0, 3: 0, 2: 0, 4: 0, 6: 0, 8: 0}
  5. Final Validation and Return: Now ans has 3 elements, which is half of the changed array's length. Therefore, assuming that a double for each element existed, our original array is [1, 3, 4].

This simplified example demonstrates how each step logically brings us closer to the reconstructed original array or leads us to conclude that reconstruction is not possible if the conditions are not met.

Solution Implementation

1from collections import Counter
2
3class Solution:
4    def findOriginalArray(self, changed: List[int]) -> List[int]:
5        # Get the length of the array
6        n = len(changed)
7        # If the length of the array is odd, we cannot form a doubled array
8        if n % 2 == 1:
9            return []
10        # Count the frequency of each element in the array
11        element_counter = Counter(changed)
12        # Sort the array to handle pairs in ascending order
13        changed.sort()
14        # Initialize the original array
15        original_array = []
16      
17        # Iterate over the sorted array
18        for number in changed:
19            # Skip the number if it has already been paired
20            if element_counter[number] == 0:
21                continue
22            # If there isn't a double of this number, we can't form a valid array
23            if element_counter[number * 2] <= 0:
24                return []
25            # Add the number to the original array and adjust the counts for the number and its double
26            original_array.append(number)
27            element_counter[number] -= 1
28            element_counter[number * 2] -= 1
29
30        # Return the original array only if it is half the size of the changed array; otherwise, return an empty array
31        return original_array if len(original_array) == n // 2 else []
32
33# Here is the correct usage of list and typing import for the List type annotation
34from typing import List
35
1import java.util.Arrays;
2
3class Solution {
4    public int[] findOriginalArray(int[] changed) {
5        // Find the length of the changed array
6        int length = changed.length;
7
8        // if the length is odd, there cannot be an original array 
9        // because the original and double elements aren't in pairs
10        if (length % 2 == 1) {
11            return new int[0];
12        }
13      
14        // Sort the array to ensure the paired items can be found easily
15        Arrays.sort(changed);
16
17        // Initialize the count array to keep track of the frequency of each number
18        int[] frequency = new int[changed[length - 1] + 1];
19        for (int num : changed) {
20            frequency[num]++;
21        }
22      
23        // Initialize the resulting array with half the length of the changed array
24        int[] result = new int[length / 2];
25        int resultIndex = 0;
26
27        // Go through the elements in changed array to find the original numbers
28        for (int num : changed) {
29            // Skip already paired numbers
30            if (frequency[num] == 0) {
31                continue;
32            }
33            // If the double value is out of the frequency array's range or already used up
34            if (num * 2 >= frequency.length || frequency[num * 2] <= 0) {
35                return new int[0];
36            }
37            // If a valid pair is found, put it in the result
38            result[resultIndex++] = num;
39            // Decrement the counts for the number and its double
40            frequency[num]--;
41            frequency[num * 2]--;
42        }
43      
44        // Check if we've successfully found the original array
45        return resultIndex == length / 2 ? result : new int[0];
46    }
47}
48
1#include <vector>
2#include <algorithm>
3
4class Solution {
5public:
6    vector<int> findOriginalArray(vector<int>& changed) {
7        int n = changed.size();
8
9        // If the size is odd, it's impossible to form an original array
10        if (n % 2 != 0) {
11            return {};
12        }
13
14        // Sort the array to make sure that for every element x, x*2 comes after x if it exists
15        sort(changed.begin(), changed.end());
16
17        // Create a frequency array that accounts for all elements in 'changed'
18        vector<int> frequency(changed.back() + 1, 0);
19        for (int x : changed) {
20            frequency[x]++;
21        }
22
23        // Initialize the vector to store the original array elements
24        vector<int> originalArray;
25
26        // Iterate over the sorted 'changed' array
27        for (int x : changed) {
28            if (frequency[x] == 0) {
29                // If the current element's count is already 0, skip it
30                continue;
31            }
32
33            if (x * 2 >= frequency.size() || frequency[x * 2] <= 0) {
34                // If there are no elements double the current or outside of the count range, return empty
35                return {};
36            }
37
38            // Decrement the count of the element and its double
39            originalArray.push_back(x);
40            frequency[x]--;
41            frequency[x * 2]--;
42        }
43
44        // If the size of formed originalArray is exactly half of the 'changed' array, return it
45        // Otherwise, return an empty vector indicating no valid original array could be formed
46        return originalArray.size() == n / 2 ? originalArray : vector<int>();
47    }
48};
49
1function findOriginalArray(changed: number[]): number[] {
2    // Check if the array length is odd; if so, there can't be an original array
3    const length = changed.length;
4    if (length % 2 !== 0) {
5        return [];
6    }
7
8    // Create a map to count occurrences of each number
9    const frequencyCounter = new Map<number, number>();
10
11    // Populate the frequency counter with the frequency of each number
12    for (const number of changed) {
13        frequencyCounter.set(number, (frequencyCounter.get(number) || 0) + 1);
14    }
15
16    // Sort the array to process pairs in order
17    changed.sort((a, b) => a - b);
18
19    // Initialize an array to store the original array
20    const originalArray: number[] = [];
21
22    // Iterate through the sorted array to find and validate pairs
23    for (const number of changed) {
24        // If the current number is already processed, skip it
25        if (frequencyCounter.get(number) === 0) {
26            continue;
27        }
28    
29        // If there is no valid pair for the current number, return an empty array
30        if ((frequencyCounter.get(number * 2) || 0) <= 0) {
31            return [];
32        }
33    
34        // Add the current number to the original array
35        originalArray.push(number);
36    
37        // Decrement the frequency of the current number and its pair
38        frequencyCounter.set(number, frequencyCounter.get(number)! - 1);
39        frequencyCounter.set(number * 2, frequencyCounter.get(number * 2)! - 1);
40    }
41
42    // If the original array's length is half of the changed array, return it; otherwise, return an empty array
43    return originalArray.length * 2 === length ? originalArray : [];
44}
45

Time and Space Complexity

Time Complexity

The time complexity of the provided code can be broken down as follows:

  1. Getting the length of changed (n = len(changed)) is an O(1) operation.
  2. Checking for even length (if n & 1) is also an O(1) operation.
  3. Creating a counter (cnt = Counter(changed)) counts the frequency of each number in changed, which is an O(n) operation.
  4. Sorting the array (changed.sort()) takes O(n log n) time.
  5. The loop runs through the sorted list of numbers (for x in changed). In the worst case, it runs n times. Each operation inside is O(1) because accessing and modifying the counter is constant time, given a good hash function.
  6. Condition checks inside the loop and counter updates are all O(1).
  7. The last return statement (return ans if len(ans) == n // 2 else []) is an O(1) operation.

The most time-consuming steps are the counter creation and the sorting. Adding the complexities, we get:

O(n) (for counting) + O(n log n) (for sorting) + O(n) (for iterating through the list) = O(n log n).

Therefore, the overall time complexity of the algorithm is O(n log n).

Space Complexity

For space complexity:

  1. Additional space is used to store the counter (cnt), which can be up to O(n) if all numbers are unique.
  2. Space for the output array (ans), which is half of the input array's size in the best case, so O(n/2) simplifies to O(n).

Hence, the overall space complexity of the algorithm is O(n).

Learn more about how to find time and space complexity quickly using problem constraints.


Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:
Question 1 out of 10

Which of the following is a min heap?


Recommended Readings

Want a Structured Path to Master System Design Too? Don’t Miss This!