题目:
给定一个正整数数列,和正整数 p,设这个数列中的最大值是 M,最小值是 m,如果 M≤mp,则称这个数列是完美数列。
现在给定参数 p 和一些正整数,请你从中选择尽可能多的数构成一个完美数列。
输入格式:
输入第一行给出两个正整数 N 和 p,其中 N(≤105)是输入的正整数的个数,p(≤109)是给定的参数。第二行给出 N 个正整数,每个数不超过 10^9。
输出格式:
在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。
输入样例:
10 8
2 3 20 4 5 1 6 7 8 9
输出样例:
8
分类:双指针
我的第一次提交:
#include<iostream>
#include<algorithm>
using namespace std;
int main() {
int n, p, j, i;
cin>>n>>p;
int a[n];
for(i=0; i<n; i++) {
scanf("%d", &a[i]);
}
long long res = 1;
sort(a, a+n);
for(i=0; i<n; i++) {
for(j=n-1; j>i&&(j-i+1)>res; j--) {
if(a[j] <= a[i]*p) {
res = res > (long long)(j-i+1) ? res : (long long)(j-i+1);
break;
}
}
}
cout<<res;
return 0;
}
提交结果:
柳神代码:
#include<iostream>
#include<algorithm>
using namespace std;
int main() {
int n, p, i, j;
scanf("%d%d", &n, &p);
int a[n];
for(i=0; i<n; i++) {
scanf("%d", &a[i]);
}
sort(a, a+n);
long long res = 0;
j = 0;
for(i=0; i<n-res; i++) {
while(j<n && a[j] <= (long long)a[i]*p) {
res = max(res, (long long)(j-i+1));
j++;
}
}
printf("%lld", res);
return 0;
}
思路改进:
柳神的第二层循环中,j只走了一遍,没有重复。而我的思路j重复走了多次,导致超时。这是一个双指针的巧妙用法。