给定一个正整数序列和另一个正整数p。 如果M≤m×p,则该序列被认为是理想序列,其中M和m分别是序列中的最大数和最小数。
现在给定一个序列和一个参数p,您应该从序列中找到尽可能多的数字以形成一个完美的子序列。
输入规格:
每个输入文件包含一个测试用例。 对于每种情况,第一行都包含两个正整数N和p,其中N(≤10-5)是序列中整数的数量,p(≤10-9)是参数。 在第二行中有N个正整数,每个不大于10 9。
输出规格:
对于每个测试用例,在一行中打印可以选择以形成理想子序列的最大整数数。
输入样例:
10 8
2 3 20 4 5 1 6 7 8 9
样本输出:
8
思路:
先将数组进行排序,然后从这个数组的每一个位置都往后遍历一遍找到满足条件的最大值,不断更新最大值。
代码一(two pointers):
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <set>
#include <map>
using namespace std;
int main()
{
int n,p,a[100100];
cin>>n>>p;
for(int i=0;i<n;i++)
cin>>a[i];
sort(a,a+n); //先排序
int z=0,y=0,ans=0; //z左 y右 ans答案
while(z<n) //左边位置小于n
{
while(a[y]<=(long long)a[z]*p && y<n) //加longlong测试点5
//满足条件时就一直往后,直到不满足了,就是本次最大值
{
ans = max(ans,y-z+1); //更新最大值
y++; //右边 继续往后
}
z++; //左边往后一个
}
cout<<ans;
return 0;
}
代码二(二分 upper_bound):
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <set>
#include <map>
using namespace std;
int main()
{
int n,p,a[100100];
cin>>n>>p;
for(int i=0;i<n;i++)
cin>>a[i];
sort(a,a+n); //排序
int ans=0;
for(int i=0;i<n;i++)
{
int position=upper_bound(a+i,a+n,(long long)a[i]*p)-a;
//找到第一个大于条件的数,然后更新最大值
ans = max(ans , position-i);
}
cout<<ans;
return 0;
}