[Leetcode]Ugly Number II

来源:互联网 发布:mac玩游戏卡 编辑:程序博客网 时间:2024/05/18 02:12

Write a program to find the n-th ugly number.

Ugly numbers are positive numbers whose prime factors only include 2, 3, 5. For example,1, 2, 3, 4, 5, 6, 8, 9, 10, 12 is the sequence of the first 10 ugly numbers.

Note that 1 is typically treated as an ugly number.

Hint:

  1. The naive approach is to callisUgly for every number until you reach the nth one. Most numbers arenot ugly. Try to focus your effort on generating only the ugly ones.
  2. An ugly number must be multiplied by either 2, 3, or 5 from a smaller ugly number.
  3. The key is how to maintain the order of the ugly numbers. Try a similar approach of merging from three sorted lists: L1, L2, and L3.
  4. Assume you have Uk, the kth ugly number. Then Uk+1 must be Min(L1 * 2, L2 * 3, L3 * 5).

Credits:
Special thanks to @jianchao.li.fighter for adding this problem and creating all test cases.

class Solution {public:    int nthUglyNumber(int n) {        priority_queue<long, std::vector<long>, std::greater<long> >Q;        long i,count = 0;        Q.push(1);        while(count != n){            i = Q.top();            Q.pop();            while(Q.size() && i== Q.top())Q.pop();            ++count;            Q.push(2*i);            Q.push(3*i);            Q.push(5*i);        }        return i;    }};

0 0
原创粉丝点击