寒假每日一题W1D3——上课睡觉

题目描述

有 N 堆石子，每堆的石子数量分别为 a1,a2,…,aN。

你可以对石子堆进行合并操作，将两个相邻的石子堆合并为一个石子堆，例如，如果 a=[1,2,3,4,5]，合并第 2,3 堆石子，则石子堆集合变为 a=[1,5,4,5]。

我们希望通过尽可能少的操作，使得石子堆集合中的每堆石子的数量都相同。

请你输出所需的最少操作次数。

本题一定有解，因为可以将所有石子堆合并为一堆。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。

每组数据第一行包含整数 N。

第二行包含 N 个整数 a1,a2,…,aN。

输出格式

每组数据输出一行结果。

数据范围

输入输出样例及解释

思路

①对于每组数据，我们先统计这组数据中石子的总和，记为sum；

②显然，最终的组数一定是sum的约数，可以整除sum，因此我们从n开始枚举这个数（因为最多有n组，最简单的情况是，什么也不用做，但是此时也需要保证n是sum的约数）；

③之后，判断枚举的组数cnt是否满足题意，即，从头开始合并石子，如果能合并成第i堆且第i堆的石子数量小于等于sum/cnt，则说明可以合并当前堆，如果超过sum/cnt，说明不合题意，继续枚举。

代码

#include <iostream>
#include <cstring>