Update super-ugly-number.cpp

XuhuiRen · Mar 29, 2018 · fcf6541 · fcf6541
1 parent 9a2ac6f
commit fcf6541
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/C++/super-ugly-number.cpp b/C++/super-ugly-number.cpp
@@ -1,4 +1,4 @@
-// Time:  O(n * logk) ~ O(n * k)
+// Time:  O(n * k)
 // Space: O(n + k)
 
 // Heap solution. (308ms)
@@ -17,7 +17,7 @@ class Solution {
             tie(uglies[i], k) = heap.top();
             heap.pop();
             ugly_by_last_prime[i] = k;
-            while (ugly_by_last_prime[++idx[k]] > k);  // worst time: O(k)
+            while (ugly_by_last_prime[++idx[k]] > k);  // average time: O(k)
             heap.emplace(uglies[idx[k]] * primes[k], k);
         }
         return uglies[n - 1];