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

P1109 学生分组

题目描述

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

输入格式

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

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

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

输出格式

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

输入输出样例 #1

输入 #1

2
10 20
10 15

输出 #1

5

说明/提示

数据范围及约定

对于全部数据，保证 1 ≤ n ≤ 50 1\le n \le 50 1≤n≤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;
}