题目大意:给出一个N个数的序列,从中选出一些数,满足以下条件:选出的数中的最大值M <= 最小值m * p,p是给定值。求出最多能选出多少个数。
思路比较直白,先排序,然后依次遍历,显然遍历到的数作为最小值,然后找到满足 <= m * p的元素的最大下标,之后就可以求出以当前元素为最小值的可能序列长度。找最大元素下标可以用二分法手动实现,不过要注意返回的是最后一个 <= 查找值的元素下标。具体而言,如果 left > right,说明没找到,并且 right就是要求的下标。但是如果找到了这个元素,即 v[mid] == value,还需要找到最后一个等于value的元素下标,比较麻烦。另外一种思路是直接用STL的 upper_bound函数,返回第一个大于查找元素位置的迭代器,对它-1就是最后一个<=查找元素的迭代器。upper_bound的调用格式如 :upper_bound(data.begin(), data.end(), value),注意是在左闭右开区间查找,返回的是迭代器,求下标的话要 - data.begin。还有一个函数 lower_bound,返回的是第一个大于等于查找元素的迭代器,这一题不适用。
另外要注意简化方式,如果把 v[i] * p > max{v}作为遍历结束的条件的话,会造成丢解,因为这样很大概率取不到最后一个元素。事实上这题不简化也不会超时,遍历完所有数字都可以。
AC代码:
二分法:
#include <vector>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef unsigned long long ull;
int search(ull value, int left, int right, vector<ull> &v)
{
while(left <= right)
{
int mid = (left + right) / 2;
if(value == v[mid])
{
while(mid < v.size() && v[mid] == value) mid++;
if(mid == v.size() - 1 && v[mid] == value) return mid;
else return mid-1;
}
else if(value < v[mid]) right = mid - 1;
else left = mid + 1;
}
return right;
}
int main()
{
ull N, p;
scanf("%lld%lld", &N, &p);
vector<ull> v(N);
for (int i = 0; i < N; ++i)
scanf("%lld", &v[i]);
sort(v.begin(), v.end());
int maxSize = 1;
for (int i = 0; i < N; ++i)
{
int size = search(v[i] * p, i, v.size() - 1, v) - i + 1;
if(maxSize < size) maxSize = size;
}
printf("%d", maxSize);
return 0;
}
upper_bound:
#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
int main()
{
ull N, p;
scanf("%lld%lld", &N, &p);
vector<ull> v(N);
for (int i = 0; i < N; ++i)
{
scanf("%lld", &v[i]);
}
sort(v.begin(), v.end());
int maxCnt = 0;
for (int i = 0; i < v.size() - maxCnt; ++i)
{
int right = upper_bound(v.begin()+i, v.end(), v[i] * p) - v.begin();
maxCnt = max(maxCnt, right - i);
}
printf("%d", maxCnt);
return 0;
}