PAT Basic1030 完美数列

版权声明：个人学习笔记记录 https://blog.csdn.net/Ratina/article/details/84547252

链接：PAT Basic1030

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

扫描二维码关注公众号，回复： 4269840 查看本文章

---

先排序，然后暴力查找最长的符合序列，但是要进行一些优化，不然依旧会超时，具体看代码。
借鉴了别人的优化方法，感觉挺巧妙的。

m*p可能会超过int的存储范围，建议用long long存储。

---

以下代码：

#include<algorithm>
#include<cstdio>
using namespace std;
int main()
{
	long long N,p;
	while(scanf("%lld%lld",&N,&p)!=EOF)
	{
		int i,j,max=0;    //max为最长序列长度
		long long ar[100005];
		for(i=0;i<N;i++)
			scanf("%lld",&ar[i]);
		sort(ar,ar+N);
		for(i=0;i<N;i++)
		{	
			long long mp=ar[i]*p; 
			for(j=max+i;j<N;j++)   //这里j=max+i，直接从现在的最长长度开始查找
			{		        //如果没有符合的，那这个max就肯定是最长的了，减少了很多不必要的循环
				if(ar[j]>mp)
					break;
				if(j-i+1>max)
					max=j-i+1;
			}
		}
		printf("%d\n",max);
	}
	return 0;
}