题目链接:传送门
思路:先对输入数组进行排序,排完存储前缀和,之和分析可以得到,对一个位置t
要让它排到p位置首先需要对自己需要加m,再判断能否大于等于a[p],可以就继续,不可以则退出,然后a[p-1]以及之前的数组不受影响,a[t]之后的也不受影响,可以加m,剩下的最多加到m + a[t]即可,然后可以证明:只要加的值小于m*v,一定存在分配策略。
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
const int INF = 0x7f7f7f7f;
typedef long long ll;
ll a[maxn];
ll sum[maxn];
vector <ll> vt;
bool cmp(int a , int b) {
return a > b;
}
int main() {
ll n , m , v , p;
ios::sync_with_stdio(0);
cin >> n >> m >> v >> p;
vt.push_back(0);
for(int i = 1 ; i <= n ; i++) {
cin >> a[i];
}
sort(a + 1 , a + n + 1, cmp);
int ans = p;
for(int i = 1 ; i <= n ; i++) {
sum[i] = sum[i - 1] + a[i];
}
for(int i = p + 1 ; i <= n ; i++) {
if(a[i] + m < a[p]){
break;
}
else {
if(m * v <= (p - 1) * m + (n - i + 1) * m+ (a[i] + m) * (i - p) - (sum[i - 1] - sum[p - 1])) {
ans ++;
}
else {
break;
}
}
}
cout << ans << "\n";
return 0;
}