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;
int a[51],l,r,tot,h,q;
int n;
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];             
		tot+=a[i];            
	}
	cin>>l>>r;                    
	if(tot<(l*n)||tot>(r*n)){
		cout<<-1;
		return 0;
	}else{
		for(int i=1;i<=n;i++){
			if(a[i]>r) h+=a[i]-r;
			if(a[i]<l) q+=l-a[i];
		}
		cout<<max(h,q);
		return 0;
		}
}