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

题目链接：传送门

思路：先对输入数组进行排序，排完存储前缀和，之和分析可以得到，对一个位置t
要让它排到p位置首先需要对自己需要加m，再判断能否大于等于a[p]，可以就继续，不可以则退出，然后a[p-1]以及之前的数组不受影响，a[t]之后的也不受影响，可以加m，剩下的最多加到m + a[t]即可，然后可以证明：只要加的值小于m*v，一定存在分配策略。

代码：

#include 

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;
}