目录
牛客_分组_枚举+二分
题目解析
C++代码
Java代码
牛客_分组_枚举+二分
分组 (nowcoder.com)
描述:
dd当上了宣传委员,开始组织迎新晚会,已知班里有nnn个同学,每个同学有且仅有一个擅长的声部,把同学们分成恰好mmm组,为了不搞砸节目,每一组里的同学都必须擅长同一个声部,当然,不同组同学擅长同一个声部的情况是可以出现的,毕竟一个声部也可以分成好几个part进行表演,但是他不希望出现任何一组的人过多,否则可能会导致场地分配不协调,也就是说,她希望人数最多的小组的人尽可能少,除此之外,对组内人员分配没有其他要求,她希望你告诉她,这个值是多少,如果无法顺利安排,请输出-1。
题目解析
暴力枚举:从小到大枚举所有可能的最大值,找到第一个符合要求的最大值。
二分优化枚举:二分出符合要求的最大值。
C++代码
#include <iostream>
#include <unordered_map>
using namespace std;
int n, m;
unordered_map<int, int> cnt; // 统计每种声部的⼈数
bool check(int x) // 判断最多⼈数为 x 时,能否分成 m 组
{
int g = 0; // 能分成多少组
for(auto& [a, b] : cnt)
{
g += b / x + (b % x == 0 ? 0 : 1);
}
return g <= m;
}
int main()
{
cin >> n >> m;
int hmax = 0; // 统计声部最多的⼈数
for(int i = 0; i < n; i++)
{
int x;
cin >> x;
hmax = max(hmax, ++cnt[x]);
}
int kinds = cnt.size();
if(kinds > m) // 处理边界情况
{
cout << -1 << endl;
}
else
{
// // 暴⼒枚举
// for(int i = 1; i <= hmax; i++) // 枚举所有的最多⼈数
// {
// if(check(i))
// {
// cout << i << endl;
// break;
// }
// }
// ⼆分解法
int l = 1, r = hmax;
while(l < r)
{
int mid = (l + r) / 2;
if(check(mid))
{
r = mid;
}
else
{
l = mid + 1;
}
}
cout << l << endl;
}
return 0;
}
Java代码
import java.util.*;
public class Main
{
public static int n, m;
public static Map<Integer, Integer> hash = new HashMap<>(); // 统计每种声部的⼈数
public static boolean check(int x) // 判断最多⼈数为 x 时,能否分成 m 组
{
int g = 0; // 统计能分成多少组
for(int a : hash.values())
{
g += a / x + (a % x == 0 ? 0 : 1);
}
return g <= m;
}
public static void main(String[] args)
{
Scanner in = new Scanner(System.in);
n = in.nextInt(); m = in.nextInt();
int hmax = 0; // 所有声部的最⼤值
for(int i = 0; i < n; i++)
{
int x = in.nextInt();
hash.put(x, hash.getOrDefault(x, 0) + 1);
hmax = Math.max(hmax, hash.get(x));
}
// 边界情况
int kinds = hash.size();
if(kinds > m)
{
System.out.println(-1);
}
else
{
// // 暴⼒解法
// for(int i = 1; i <= hmax; i++)
// {
// if(check(i))
// {
// System.out.println(i);
// break;
// }
// }
// ⼆分解法
int l = 1, r = hmax;
while(l < r)
{
int mid = (l + r) / 2;
if(check(mid))
{
r = mid;
}
else
{
l = mid + 1;
}
}
System.out.println(l);
}
}
}