4366. 上课睡觉
https://www.acwing.com/problem/content/description/4369/
思路
我们先分析所要的结果,每堆石子数量相同(记为 a a a ),因此每堆石子的数量一定是石子总数的约数(记为 S S S )。假设我们找到每堆石子最终数量为 a a a,则共有 S / a S/a S/a 堆,所需合并次数(记为 n u m num num)为 n − S / a n-S/a n−S/a。则要使操作次数最小,即 n u m num num 最小,则 S / a S/a S/a 要最大, a a a 则要最小。我们可以枚举 S / a S/a S/a 或者 a a a。( S / a S/a S/a 和 a a a 都为 S S S 的约数)
C++代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
int n, T;
int a[N];
int b[N];
bool check(int ave) {
// ave 为每堆有多少个
int cnt = 0, sum = 0;
for (int i = 0; i < n; i ++) {
sum += a[i];
if (sum == ave) sum = 0;
if (sum > ave) return false;
}
return true;
}
int main()
{
scanf("%d", &T);
while(T--) {
scanf("%d", &n);
int sum = 0;
for (int i = 0; i < n; i ++) {
scanf("%d", &a[i]);
sum += a[i];
}
int cnt = 0;
for (int i = n; i >= 1; i --)
if (sum % i == 0) {
b[cnt ++] = i;
}
int i = 0;
for (; i < cnt; i ++) {
if (check(sum / b[i])) {
break;
}
}
printf("%d\n", n - b[i]);
}
}