有 N堆石子,每堆的石子数量分别为 a1,a2,…,aN。
你可以对石子堆进行合并操作,将两个相邻的石子堆合并为一个石子堆,例如,如果 a=[1,2,3,4,5],合并第 2,3堆石子,则石子堆集合变为 a=[1,5,4,5]。
我们希望通过尽可能少的操作,使得石子堆集合中的每堆石子的数量都相同。
请你输出所需的最少操作次数。
本题一定有解,因为可以将所有石子堆合并为一堆。
输入格式
第一行包含整数 T,表示共有 TT组测试数据。
每组数据第一行包含整数 NN。
第二行包含 N 个整数 a1,a2,…,aN。
输出格式
每组数据输出一行结果。
数据范围
1≤T≤10
1≤N≤1e5
0≤ai≤1e6
ai之和≤1e6
每个输入所有 N之和不超过 1e5。
输入样例:
3
6
1 2 3 1 1 1
3
2 2 3
5
0 0 0 0 0
输出样例:
3
2
0
样例解释
第一组数据,只需要用 33 个操作来完成:
1 2 3 1 1 1
-> 3 3 1 1 1
-> 3 3 2 1
-> 3 3 3
第二组数据,只需要用 22 个操作来完成:
2 2 3
-> 2 5
-> 7
第三组数据,我们什么都不需要做。
这道题之前在codeforces见到过,当时就不会写,然后后来也没有看题解,今天每日一题上面遇到了,看了视频讲解,觉得还是很好理解的:
首先分析一下题目的意思,限制条件:相邻两堆才可以合并,已知:每堆石子的数量ai大于等于0,用尽可能少的操作使得每堆石子数量相同。假设所有的石子总数量为sum,合并过后共有d堆,每堆有x个,那么一定会满足的式子是sum%d==0,sum%x==0,sum/d==x,他们满足一个整除的关系,所以我们可以通过枚举堆数来确定最后形成多少堆满足要求,当然最后的堆数越多代表操作的次数越少,那么堆数d就从n开始枚举到1;然后题目的一个条件就是只有相邻的两堆才可以合并,这表示的就相当于你把原来的那些石子堆分成了若干份,每一份石子的数量都等于x,这样才符合要求,也就是下边代码中的check函数,判断当你堆数是d的时候,每一堆应该是sum/d,原本的石子堆可以形成每一份都是sum/d才符合要求。大致思路是这样的。
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
int n,w[N];
bool check(int x)
{
for(int i=0,s=0;i<n;i++)
{
s+=w[i];
if(s>x) return false;//没加新的一堆之前,小于x,加了新的一堆之后又大于x,那肯定就是不能形成x
else if(s==x) s=0;
}
return true;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
int sum=0;
for(int i=0;i<n;i++)
{
scanf("%d",&w[i]);
sum+=w[i];
}
for(int i=n;i;i--)//枚举最后的堆数,最多是n堆,最少是1堆
{
if(sum%i==0 && check(sum/i))
{
printf("%d\n",n-i);
break;
}
}
}
return 0;
}