1030. 完美数列(25)

1030. 完美数列(25)

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CAO, Peng

给定一个正整数数列，和正整数p，设这个数列中的最大值是M，最小值是m，如果M <= m * p，则称这个数列是完美数列。

现在给定参数p和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入格式：

输入第一行给出两个正整数N和p，其中N（<= 10⁵）是输入的正整数的个数，p（<= 10⁹）是给定的参数。第二行给出N个正整数，每个数不超过10⁹。

输出格式：

在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。

输入样例：

10 8
2 3 20 4 5 1 6 7 8 9

输出样例：

8

#include <stdio.h>
#define MAX 100000
void Merge(int* data, int* tmp, int lBegin, int lEnd, int rEnd) { //归并部分
	int rBegin = lEnd + 1;
	int i = lBegin, j = rBegin, k = lBegin;
	while(i <= lEnd && j <= rEnd)
		if(data[i] < data[j])
			tmp[k++] = data[i++];
		else
			tmp[k++] = data[j++];
	while(i <= lEnd)
		tmp[k++] = data[i++];
	while(j <= rEnd)
		tmp[k++] = data[j++];
//	memcpy(data + lBegin, tmp + lBegin, (rEnd - lBegin + 1) * sizeof(int));
	for (int k = lBegin; k <= rEnd; ++k)
		data[k] = tmp[k];
}
void Sort(int* data, int* tmp, int left, int right) { //归并排序
	if(right > left) {
		int mid = (left + right) >> 1;
		Sort(data, tmp, left, mid);
		Sort(data, tmp, mid + 1, right);
		Merge(data, tmp, left, mid, right);
	}
}
int main() {
	freopen("in", "r", stdin);
	int data[MAX], tmp[MAX];
	int n, p;
	scanf("%d%d", &n, &p);
	for(int i = 0; i < n; ++i)
		scanf("%d", &data[i]);
	Sort(data, tmp, 0, n - 1);		//排序
	//计算每一项作为数列开头时能允许的最大数列
	int max = 0;	//记录当前允许的最长数列
	for(int i = 0; i < n - max; ++i)	//最后不剩max项时不需要再计算了
		while(i + max < n && (long long)data[i + max] <= (long long)data[i] * p)	//直接跳到第max+1项，忽略前面的判断
			++max;
	printf("%d", max);

	return 0;
}

总结：

1.先从小到大排序数列，循环以每一项为数列头计算最长允许的子序列；考虑到只有大于max的值才有计算的意义，每一轮循环中，不需要每次从头开始遍历，从当前子序列的第max+1项开始，也不需要计算最后的max个循环，否则有几组数据会超时。

2.归并排序还是不熟练，把递归函数Sort()中的参数left写成0，OJ居然只是有两组数据提示超时，还以为是哪部分效率不高，调试了一晚上才发现。