2 3 5 7作为因子构成的数字序列——笔试题

类似leetcode的丑数II 

链接：https://leetcode-cn.com/problems/ugly-number-ii/

说回这个题

而这个是只有2 3 5 7四种因子，不包含本身，即序列为4 6 8 9 10 ...

输入正整数n

输出序列中第n个数 （可能很大，对1e9+7取模）

用dp三指针实现，cpp代码如下

#include <bits/stdc++.h>

using namespace std;

int mo = 1e9+7;

int getNum(int n){
    vector<int> dp(n+10);
    dp.at(0) = 1;
    int p2,p3,p5,p7;
    p2 = p3 = p5 = p7 = 0;
    for(int i=1;i<n+10;++i){
        dp.at(i) = min(min(min(2 * dp.at(p2), 3 * dp.at(p3)), 5 * dp.at(p5)),7 * dp.at(p7)) % mo;    //计算数组, 直接取模
        if(dp.at(i) == 2 * dp.at(p2))
            ++p2;
        if(dp.at(i) == 3 * dp.at(p3))
            ++p3;
        if(dp.at(i) == 5 * dp.at(p5))
            ++p5;
        if(dp.at(i) == 7 * dp.at(p7))
            ++p7;
    }
    for(vector<int>::iterator iter = dp.begin();iter!=dp.end(); ){      //删除2 3 5 7四个原因子
    if( *iter == 2 || *iter == 3 || *iter == 5 || *iter == 7){
          iter = dp.erase(iter);
    }else
            iter ++ ;
    }
    return dp.at(n);
}

int main()
{
    int n;
    cin>>n;
    cout<<getNum(n);
    return 0;
}

测试输出一下序列，正确。

可惜昨晚笔试的时候没想出来