AtCoder Grand Contest 041 B - Voting Judges (前缀和)

题目链接:传送门

思路:先对输入数组进行排序,排完存储前缀和,之和分析可以得到,对一个位置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;
}

猜你喜欢

转载自blog.csdn.net/qq_39475280/article/details/103775468