洛谷：P1109 学生分组 (贪心、模拟)

题目描述

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

输入格式

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

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

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

输出格式

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

输入输出样例

输入

2
10 20
10 15

输出 

5

说明/提示

数据范围及约定

对于全部数据，保证 1≤n≤50。

题目翻译：

给出一个包含了 n 个数的数列 a，你可以进行一种操作，这种操作支持：对于任意的i,j∈{1,2,3,...,n}，可以让ai​ 变大为ai​+1，并且aj​ 变小为 aj​−1。问的是最多需要多少步操作，才能使对于任意的 i∈{1,2,3,4,..,n}，都有 l≤ai​≤r。

这个应该足够容易理解了吧。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
int n, L, R;
int arr[100];
int a, b, c, d; //a:可以+ b:可以- c:必须+ d:必须-
int main()
{
	cin >> n;
	for (int i = 0; i < n; i++)cin >> arr[i];

	cin >> L >> R;
	for (int i = 0; i < n; i++)
	{
		if (arr[i] <= L) //必须+
		{
			int t = L - arr[i];
			c += t;
			a += R-L; //因为我们已经确定它必须加 L - arr[i],那么我们默认他已经加到了L，此时就还可以加 R-L
		}
		if (arr[i] >= R) //必须-
		{
			int t = arr[i] - R;
			d += t;
			b += R-L; //因为我们已经确定它必须减 arr[i] - R,那么我们默认他已经减到了L，此时就还可以减 R-L
		}
		if (arr[i] >= L && arr[i] <= R) //既可以+也可以-
		{
			int t1 = arr[i] - L; //可以-的人数
			int t2 = R - arr[i]; //可以+的人数
			a += t2; 
			b += t1;
		}
	}
	if (c > d) //如果最后必须加的人数比必须减的人数多
	{
		c -= d;  //就把其减掉，这时必须加的人数就变为c-d了，必须减的人数已经分配到必须加的人数里面（必须减的人数变为0），不用考虑了
		if (c <= b)
		{
			cout << c + d << endl; //此时如果必须加的人数任然小于可以减的人数，就说明还可以继续分配到这些队伍里面
		}
		else
			cout << "-1\n"; //否则输出-1
	}
	else
	{
		d -= c;
		if (d <= a)
		{
			cout << d + c << endl;
		}
		else
			cout << "-1\n";
	}
	
	return 0;
}