牛客小白月赛80C-D

https://ac.nowcoder.com/acm/contest/67730

C又放学辣（简单）

思路：数据小，枚举剩下的人数一共有0-n种，就是枚举人数最多的班级的最少的可能人数，   

比如 人数最多的班级有a个人，那么其他的班级就只能小于等于a，那么用班级的原来人数减去a求

和,就可以得到人数最多的班级的最少的可能人数为a时离开的人数，判断其满不满足k

#include<bits/stdc++.h>
using namespace std;
#define Max 0x7fffffff
#define  LL long long
#define  ULL unsigned long long
#define N 200005
int a[N];
int main() {
    int m, n, K; scanf("%d%d%d", &n, &m, &K);
    for (int i = 0; i < n; i++) {
        int t; scanf("%d", &t);
        a[t]++;
    }
    for (int i = 1; i <= m; i++) {
        if (n - a[i] < K) {
            printf("-1 ");
            continue;
        }
        for (int j = 0; j <= n; j++) {//枚举可能剩下的人数
            int num = 0;
            for (int k = 1; k <= m; k++) {
                if (k == i)continue;//如果留堂就下一个
                num += max(0, a[k] - j);//num统计每个班离开的人数，
            }
            if (num <= K) {//如果离开的人数合法那么输出剩下的人数（因为是从最小的情况枚举）
                cout << j << (" \n"[i == m]);
                break;
            }
        }
    }
    
}

 D 

思路：二分枚举，用前缀和计算（orz）

#include<bits/stdc++.h>
using namespace std;
#define Max 0x7fffffff
#define  LL long long
#define  ULL unsigned long long
#define N 200005
int a[N],sum1[N],sum2[N];
int calc(int x, int i) {
    return (sum2[x] - (a[i] >= x) * a[i]) - (sum1[x] - (a[i] >= x)) * x;//如果a[i]的人数也是大于x则要减去，a[i]是留堂的
}                                                                       //这个函数就是计算离校人数的，x是剩下的人数
int main() {
    int m, n, K; scanf("%d%d%d", &n, &m, &K);
    for (int i = 0; i < n; i++) {
        int t; scanf("%d", &t);
        a[t]++;
    }
    for (int i = 1; i <=m; i++) {
        sum1[a[i]]++;//统计人数为a[i]的班级有多少个
        sum2[a[i]] += a[i];//人数相同班级的和
    }
    for (int i = n - 1; i >= 0; i--) {
        sum1[i] += sum1[i + 1];//班级人数大于等于i的班级数量和
        sum2[i] += sum2[i + 1];//班级人数大于等于i的人数和
    }
    for (int i = 1; i <= m; i++) {
        if (n - a[i] < K) {
            cout << -1 << (" \n"[i == m]);
            continue;
        }
        int l = 0, r = n;
        while (l < r) {
            int mid = (l + r) >> 1;

            if (calc(mid, i) <= K)
                r = mid;
            else
                l = mid + 1;
        }
        printf("%d", l);
    }
}