【C++ 真题】P1109 学生分组

P1109 学生分组

题目描述

有 $n$ 组学生，给出初始时每组中的学生个数，再给出每组学生人数的上界 $R$ 和下界 $L(L\le R)$，每次你可以在某组中选出一个学生把他安排到另外一组中，问最少要多少次才可以使 $N$ 组学生的人数都在 $[L,R]$ 中。

输入格式

第一行一个整数 $n$，表示学生组数；

第二行 $n$ 个整数，表示每组的学生个数；

第三行两个整数 $L,R$，表示下界和上界。

输出格式

一个数，表示最少的交换次数，如果不能满足题目条件输出 $-1$。

输入输出样例 #1

输入 #1

2
10 20
10 15

输出 #1

5

说明/提示

数据范围及约定

对于全部数据，保证 $1\le n\le 50$。

题解

#include "bits/stdc++.h"
using namespace std;
const int N = 1e4+2;
int n, L, R, g[N], sum = 0, a[N], ans1, ans2;
int main(){
	cin>>n;
	for(int i=1;i<=n;++i){
		cin>>g[i];
		sum += g[i];
	}
	cin>>L>>R;
	if(sum < L*n || sum > n*R){
		cout<<"-1"<<endl;
		return 0;
	}
	for(int i=1;i<=n;++i){
	    if(g[i] > R) a[i] = g[i]-R;
		else if(g[i] < L) a[i] = g[i]-L;
		
		if(a[i] < 0) ans1 += a[i];
		else if(a[i] > 0) ans2 += a[i];
	}
	cout<<max(abs(ans1), ans2)<<endl;
	return 0;
}