这个题目虽然有两种方法,但是毫无疑问,双指针法要更加简单一些,也可能是自己对二分法还不是很熟悉。
双指针法:
#include<cstdio>
#include<algorithm>
using namespace std;
int main()
{
int n, p;
scanf("%d %d", &n, &p);
int a[100010];
for (int i = 0; i < n; i++)
{
scanf("%d", &a[i]);
}
sort(a, a + n);//默认由小到大
int i=0, j=0,count=0;
while (i<n&&j<n)
{
while (j < n&&a[j] <= (long long)a[i] * p)
{
count = max(count, j - i + 1);
j++;
}
i++;
}
printf("%d", count);
return 0;
}
二分法:
#include<cstdio>
#include<algorithm>
using namespace std;
int n, p;
int a[100010];
int bsearch(int i, long long x)//找到第一个大于x的位置
{
if (a[n - 1] <= x) return n;
int l = i + 1, k = n - 1, mid;
while (l < k)
{
mid = (l + k) / 2;
if (a[mid] <= x) l = mid + 1;
else k = mid;
}
return l;
}
int main()
{
scanf("%d %d", &n, &p);
for (int i = 0; i < n; i++)
{
scanf("%d", &a[i]);
}
sort(a, a + n);//默认由小到大
int ans = 1;
for (int i = 0; i < n; i++)
{
int j = bsearch(i, (long long)a[i] * p);//这个地方有一个小坑点,我定义一个long long型的l=a[i]*p替换进去,会显示有一个点错误
ans = max(ans, j - i);
}
printf("%d", ans);
return 0;
}
还有一个使用upper_bound函数,省去写二分算法的方法:
#include<cstdio>
#include<algorithm>
using namespace std;
int n, p;
int a[100010];
int main()
{
scanf("%d %d", &n, &p);
for (int i = 0; i < n; i++)
{
scanf("%d", &a[i]);
}
sort(a, a + n);//默认由小到大
int ans = 1;
for (int i = 0; i < n; i++)
{
int j =upper_bound(a+i+1,a+n,(long long)a[i]*p)-a;
ans = max(ans, j - i);
}
printf("%d", ans);
return 0;
}