The idea is to use an array to store the ugly numbers and use three pointers for 2, 3, and 5 to calculate the next potential ugly numbers. We then choose the minimum of these numbers to be the next ugly number and appropriately move the pointers.
Time Complexity: O(n)
, where n
is the number of ugly numbers to generate.
Space Complexity: O(n)
, for storing the ugly numbers array.
1using System;
2
3public class Solution {
4 public int NthUglyNumber(int n) {
5 int[] ugly = new int[n];
6 ugly[0] = 1;
7 int i2 = 0, i3 = 0, i5 = 0;
8 int next_2 = 2, next_3 = 3, next_5 = 5;
9
10 for (int i = 1; i < n; i++) {
11 int next_ugly = Math.Min(next_2, Math.Min(next_3, next_5));
12 ugly[i] = next_ugly;
13
14 if (next_ugly == next_2) next_2 = ugly[++i2] * 2;
15 if (next_ugly == next_3) next_3 = ugly[++i3] * 3;
16 if (next_ugly == next_5) next_5 = ugly[++i5] * 5;
17 }
18 return ugly[n - 1];
19 }
20
21 public static void Main(string[] args) {
22 Solution sol = new Solution();
23 int n = 10;
24 Console.WriteLine(sol.NthUglyNumber(n)); // Output: 12
25 }
26}
27
The C# solution mirrors the approach used in the other languages, storing ugly numbers in an array with three pointers for multiples of 2, 3, and 5. It identifies each subsequent ugly number and updates the array.
This method involves using a min-heap to manage the sequence of potential ugly numbers. We start with 1 in the min-heap and repeatedly extract the smallest element, multiplying it by 2, 3, and 5 to generate new candidates, which are then inserted back into the heap. Duplicate entries are avoided by using a set for tracking which numbers have been added to the heap.
Time Complexity: O(n log n)
, primarily due to heap operations.
Space Complexity: O(n)
, for data structures that might store up to n numbers.
1import heapq
2
3def nthUglyNumber(n):
4 min_heap = [1]
5 seen = {1}
6
7 for _ in range(n):
8 curr_ugly = heapq.heappop(min_heap)
9 for factor in [2, 3, 5]:
10 new_ugly = curr_ugly * factor
11 if new_ugly not in seen:
12 seen.add(new_ugly)
13 heapq.heappush(min_heap, new_ugly)
14 return curr_ugly
15
16n = 10
17print(nthUglyNumber(n)) # Output: 12
18
The Python solution leverages a min-heap for ordering ugly numbers. The smallest number is removed in each iteration and multiplied by 2, 3, and 5 to yield potential new ugly numbers. Tracking entries in a set prevents duplicates in the heap.