题解_AcWing_2023寒假每日一题_4366_上课睡觉_

题目链接

题目描述

n堆石子, 每次可以合并相邻的两堆石子,
求最少合并次数, 使得合并后每堆石子的个数相同;

分析与思路

• 题目一定有解, 因为可以最终合并为1堆;
• 最终石子的堆数cnt一定能够整除石子总数sum, 即 $cnt|sum$, 这就要考虑sum的约数;
• 关于1~1e6范围的数, 约数最多的数为720720, 有240个约数;
  而在int范围内的数, 约数最多的数有1600个约数;
• 思路即, 遍历sum的每个约数cnt, 尝试将之合并为cnt堆, 而求最小操作次数即求最大的cnt;
• 每次检验cnt是否满足要求为 $O(n)$, 而再乘上 240, 即 1e5 * 240, 复杂度满足要求;
• 至于求约数, 这里直接遍历即可, 并不会增加复杂度;
• 关于检测cnt是否满足要求(将石子合并为cnt堆使得每堆的石子数相同),
  从最左侧的石子堆开始合并, 如果没有达到 sum / cnt, 就继续合并成一堆;
  如果当前堆刚好合并成 sum / cnt, 则新开启一堆;
  如果当前堆石子个数超过了 sum / cnt, 则cnt一定不能满足要求;

代码

#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int a[N];
int n;

bool check(int x)
{
    int sum = 0;
    for(int i = 0; i < n; i ++)
    {
        sum += a[i];
        if(sum > x) return false;
        else if(sum == x) sum = 0;
    }
    return true;
}

int main()
{
    int T;
    cin >> T;
    while(T --)
    {
        cin >> n;
        int sum = 0;
        for(int i = 0; i <  n; i ++)
        {
            cin >> a[i];
            sum += a[i];
        }
        for(int i = n; i >= 1; i --)
        {
            if(sum % i == 0 && check(sum / i))
            {
                cout << n - i << endl;
                break;
            }
        }
    }

    
    return 0 ;
}

总结