每日一题----2022.12.29

题目

---

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

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

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

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

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

输入格式

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

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

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

输出格式

每组数据输出一行结果。

数据范围

1≤T≤10
1≤N≤10^5
0≤ai≤10^6
(a1+a2+...+aN)≤10^6
每个输入所有 N 之和不超过 10^5

---

样例

---

---

题解 

---

#include <iostream>
using namespace std;
const int N = 100010;

int n;
int w[N];

bool check(int cnt)//检查是否满足条件
{
    for (int i = 0, s = 0; i < n; i ++ )
    {
        s += w[i];
        if (s > cnt) return false;
        if (s == cnt) 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 -- )//i表示最终堆数
            if (sum % i == 0 && check(sum / i))
            {
                printf("%d\n", n - i);//合并次数=总堆数-最终堆数
                break;
            }
    }
    return 0;
}

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
const int N = 100010;

using namespace std;
bool check(int i,int n,int a[])//检查是否能满足条件
{
	int p = 0;
	for (int k = 0; k < n; k++)
	{
		p += a[k]; 
		if (p == i)p = 0;
		if (p > i)return false;
	}
	return true;
}
int main() {
	int m, n;
	cin >> m;
	while (m--)
	{
		int sum = 0;
		int a[N]={0};
		cin >> n;
		for (int i = 0; i < n; i++)
			cin >> a[i], sum += a[i];
		for (int i = 1; i; i++)//i表示最终每堆的石子个数
		{
			
			if (sum % i == 0 && check(i, n,a))
			{
				if (sum == 0) { cout << 0<<endl; break; }//全为0的情况分类讨论
				cout << n - sum / i << endl; break;//不全为0的情况
			}
		}


	}

注意事项

---

1.思路很重要，连续的两堆石子才能合并表明可以通过遍历累加看是否符合要求

2.时间复杂度要注意，以防超时