USACO Humble Number

参考：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;
}