剑指 Offer 49. 丑数（三指针法）

一、263. 丑数

263. 丑数

class Solution {
public:
    bool isUgly(int n) {
        if(n <= 0){
            return false;
        }
        while(n % 2 == 0){
            n >>= 1;
        }
        while(n % 3 == 0){
            n /= 3;
        }
        while(n % 5 == 0){
            n /= 5;
        }
        return n == 1;
    }
};

二、264. 丑数 II

264. 丑数 II

1. 三指针法

class Solution {
public:
    int nthUglyNumber(int n) {
        int idx2 = 0;
        int idx3 = 0;
        int idx5 = 0;
        vector<int> nums(n, 0);
        nums[0] = 1;  // 第一个丑数为1

        for(int i = 1; i < n; i++){
        	// 由数组里的丑数生成新的丑数，将最小的丑数存入数组
            nums[i] = min(nums[idx2] * 2, min(nums[idx3] * 3, nums[idx5] * 5));
            // 由哪个idx生成新的丑数，哪个idx往后移，用于下一次生成丑数
            // 三个if并列，用于去重
            if(nums[i] == nums[idx2] * 2){
                idx2++;
            }
            if(nums[i] == nums[idx3] * 3){
                idx3++;
            }
            if(nums[i] == nums[idx5] * 5){
                idx5++;
            }
        }
        return nums[n-1];
    }
};

2. set排序且去重

set可以排序，我们用set保存已经生成的丑数，取出最小的丑数分别乘以2、3、5就可以得到新的丑数，存入set后自动排序，第n轮循环删除最小的丑数，即是第n个丑数

class Solution {
public:
	int nthUglyNumber(int n) {
        set<double> s;  // set 是有序的，且不重复
        double answer = 1;
        for (int i = 1; i < n; i++) {
            s.insert(answer * 2);
            s.insert(answer * 3);
            s.insert(answer * 5);
            answer = *s.begin();
            s.erase(answer);
        }
        return answer;
    }	
};