参考:http://hi.baidu.com/ritchie_wang/item/34431c3ceb7d54bc134b1492
思路:
我们在数组hum中计算出前n个丑数。为了实现起来更简单,我们把1也作为一个丑数,算法也要因此略微调整一下。
当我们已知前k个丑数,想得到第k+1个,我们可以这样做:对于每个质数p,寻找最小的丑数h,使得 h * p 比上一个丑数大。取我们找到的 h * p 中最小的一个,它就是下一个丑数。
为了使搜索更快,我们可以为每个质数维护一个索引“pindex”表示每个质数已经乘到了哪个丑数,每次都从那里开始,而不是再从头再来。
最后成功的代码:
/*
ID: wangxin12
PROG: humble
LANG: C++
*/
#include <iostream>
#include <vector>
#include <fstream>
using namespace std;
#define MAX 999999999
int K, N; // K <= 100, N <= 100000
//vector<long> humble;
//vector<long> prime;
long int humble[100005];
int prime[100];
int main() {
ifstream fin("humble.in");
ofstream fout("humble.out");
humble[0] = 1;
int i, j;
fin>>K>>N;
for(i = 0; i < K; i++) {
fin>>j;
prime[i] = j;
}
//记载每个prime数乘到了哪里,防止重复计算
long * pindex = new long[K + 1];
for(i = 0; i <= K; i++)
pindex[i] = 0;
long int start;
for(i = 1; i <= N; i++) {
humble[i] = 0x7fffffff;
for(j = 0; j < K; j++) {
start = pindex[j];
if(i > 0)
while(prime[j] * humble[start] <= humble[i - 1])
start++;
pindex[j] = start;
if(prime[j] * humble[start] < humble[i])
humble[i] = prime[j] * humble[start];
}
}
fout<<humble[N]<<endl;
delete[] pindex;
return 0;
}
我的第一版代码,在test 5超时,完全不能理解为什么
/*
ID: wangxin12
PROG: humble
LANG: C++
*/
#include <iostream>
#include <vector>
#include <fstream>
using namespace std;
#define MAX 999999999
int K, N; // K <= 100, N <= 100000
vector<long> humble;
vector<long> prime;
int main() {
ifstream fin("humble.in");
ofstream fout("humble.out");
humble.push_back(1); //humble[0] = 1
int i, j;
fin>>K>>N;
for(i = 1; i <= K; i++) {
fin>>j;
prime.push_back(j);
}
//记载每个prime数乘到了哪里,防止重复计算
long * pindex = new long[prime.size()];
for(i = 0; i < prime.size(); i++)
pindex[i] = 0;
while(humble.size() < N + 1) {
long small = MAX;
for(j = 0; j <= prime.size() - 1; j++) {
long start = pindex[j];
while(humble[start] * prime[j] <= humble[humble.size() - 1])
start++;
pindex[j] = start;
if(humble[start] * prime[j] < small)
small = humble[start] * prime[j];
}
if(small < MAX)
humble.push_back(small);
}
fout<<humble[N]<<endl;
delete[] pindex;
return 0;
}