洛谷 P1109 学生分组

洛谷 P1109 学生分组

---

题目

题目描述

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

输入输出格式

输入格式：
第一行一个整数N，表示学生组数； n<=50

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

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

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

输入输出样例

输入样例#1：

2
10 20
10 15

输出样例#1：

5

---

题解

首先，确认总和是不是在[l*n,r*n]之间，然后再取人数小于l的组到l所需的人数和人数大于r的组到r所需的人数的最大值即答案

---

代码

#include<cstdio>
using namespace std;

int a[55],n,p,q,tot,l,r;

int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    scanf("%d%d",&l,&r);
    tot=0;
    for (int i=1;i<=n;i++) tot+=a[i];
    if (tot<l*n||tot>r*n){
        printf("-1");
        return 0;
    }
    p=0;q=0;
    for (int i=1;i<=n;i++){
        if (a[i]>r) p+=a[i]-r;
         else if (a[i]<l) q+=l-a[i];
    }
    printf("%d",p>q?p:q);
    return 0;
}