PAT甲级 1085 Perfect Sequence (25 分)

\quad题目看起来很简单，但有两个坑。一是p虽然没超过int范围，但中间过程m*p会超int，所有在p和中间变量定义时应该用long long；二是进行优化，具体的优化实在统计满足条件的数字个数时从i+res后开始即可

#include <bits/stdc++.h>
using namespace std;

int main(int argc, char const *argv[])
{
	int n;
	long long p;
	scanf("%d%lld", &n, &p);
	long long a[n];
	for (int i = 0; i < n; ++i)
	{
		scanf("%lld", &a[i]);
	}

	sort(a, a+n);
	int res = 0, count;
	for (int i = 0; i < n; ++i)
	{
		long long upper = a[i]*p;
		count = 0;
		// 因为是为了找最大的result，所以下一次j只要从i的result个后面开始找就行了
		// 这是一个很关键的优化，否则会超时
		for (int j = i+res; j < n; ++j)
		{
			if(a[j]>upper) break; 
			count = j-i+1;
		}
		if(count>res) res=count;
	}

	printf("%d\n", res);
	return 0;
}