LeetCode 2154. Keep Multiplying Found Values by Two Solution in Java, C++, Python & More | Explanation + Code

Description

You are given an array of integers nums. You are also given an integer original which is the first number that needs to be searched for in nums.

You then do the following steps:

1. If original is found in nums, multiply it by two (i.e., set original = 2 * original).
2. Otherwise, stop the process.
3. Repeat this process with the new number as long as you keep finding the number.

Return the final value of original.

 

Example 1:

Input: nums = [5,3,6,1,12], original = 3
Output: 24
Explanation: 
- 3 is found in nums. 3 is multiplied by 2 to obtain 6.
- 6 is found in nums. 6 is multiplied by 2 to obtain 12.
- 12 is found in nums. 12 is multiplied by 2 to obtain 24.
- 24 is not found in nums. Thus, 24 is returned.

Example 2:

Input: nums = [2,7,9], original = 4
Output: 4
Explanation:
- 4 is not found in nums. Thus, 4 is returned.

 

Constraints:

• 1 <= nums.length <= 1000
• 1 <= nums[i], original <= 1000

Solutions

Solution 1: Hash Table

We use a hash table s to record all the numbers in the array nums.

Next, starting from original, if original is in s, we multiply original by 2 until original is not in s anymore, then return original.

The time complexity is O(n), and the space complexity is O(n). Here, n is the length of the array nums.

PythonJavaC++GoTypeScriptRust

class Solution: def findFinalValue(self, nums: List[int], original: int) -> int: s = set(nums) while original in s: original <<= 1 return original(code-box)

class Solution { public int findFinalValue(int[] nums, int original) { Set<Integer> s = new HashSet<>(); for (int num : nums) { s.add(num); } while (s.contains(original)) { original <<= 1; } return original; } }(code-box)

class Solution { public: int findFinalValue(vector<int>& nums, int original) { unordered_set<int> s(nums.begin(), nums.end()); while (s.contains(original)) { original <<= 1; } return original; } };(code-box)

func findFinalValue(nums []int, original int) int { s := map[int]bool{} for _, x := range nums { s[x] = true } for s[original] { original <<= 1 } return original }(code-box)

function findFinalValue(nums: number[], original: number): number { const s: Set<number> = new Set([...nums]); while (s.has(original)) { original <<= 1; } return original; }(code-box)

impl Solution { pub fn find_final_value(nums: Vec<i32>, original: i32) -> i32 { use std::collections::HashSet; let s: HashSet<i32> = nums.into_iter().collect(); let mut original = original; while s.contains(&original) { original <<= 1; } original } }(code-box)