基础 贪心专题 第5题 学生分组 WA 90分

常规
1 
#include<iostream>
#include<algorithm>
using namespace std;
int main(){
	int n,a[55],l,r;
	cin>>n;
	long long sum=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		sum+=a[i];
	}
	cin>>l>>r;
	if(sum/n>r||sum/n<l){
		cout<<"-1";
		return 0;
	}
	sort(a+1,a+n+1);
	int s=0;
	for(int i=1;i<=n;i++){
		if(a[i]<l){
			int k=l-a[i];
			for(int j=n;j>i;j--){
				if(a[j]>=l&&a[j]-l>=k){
					a[j]-=k;
					a[i]+=k;
					s+=k;
					break;
				}
				else if(a[j]>=l){
					int k1=a[j]-l;
					a[j]=l;
					a[i]+=k1;
					s+=k1;
				}
			}
		}
		else if(a[i]>r){
			int k=a[i]-r;
			for(int j=1;j<=n;j++){
				if(a[j]<=r&&r-a[j]>=k){
					a[j]+=k;
					a[i]-=k;
					s+=k;
					break;
				}
				else{
					int k1=r-a[j];
					a[j]=r;
					a[i]-=k1;
					s+=k1;
				}
			}
		}
	} 
	cout<<s;
	return 0;
}
4 个赞
2

思路可以，我帮你简化一下

2 个赞
3

20~53改成：

		if(a[i]<l){
			b+=(l-a[i]);
		}
		if(a[i]>r) c+=(a[i]-r);
2 个赞
4

别忘了定义b=0，c=0
最后输出max(b,c)

2 个赞