LeetCode 数组的均值分割

最新推荐文章于 2024-08-03 16:34:15 发布

hestyle

最新推荐文章于 2024-08-03 16:34:15 发布

阅读量2.8k

点赞数

分类专栏： LeetCode 深度、广度优先搜索数学

本文链接：https://blog.csdn.net/qq_41855420/article/details/90296879

版权

给定整数数组A，目标是将其元素移动到B和C两个空数组中，使B和C的平均值相等。讨论了解题思路，包括利用平均数关系进行剪枝，并探讨搜索技术解决子数组和的问题。

摘要由CSDN通过智能技术生成

给定的整数数组 A ，我们要将 A数组中的每个元素移动到 B数组或者 C数组中。（B数组和C数组在开始的时候都为空）

返回true ，当且仅当在我们的完成这样的移动后，可使得B数组的平均值和C数组的平均值相等，并且B数组和C数组都不为空。

示例:

输入: 
[1,2,3,4,5,6,7,8]
输出: true
解释: 我们可以将数组分割为 [1,4,5,8] 和 [2,3,6,7], 他们的平均值都是4.5。

注意:

A 数组的长度范围为 [1, 30].
A[i] 的数据范围为 [0, 10000].

$\color{blue}思路分析：$ 之前有一道题是将一个数组分成两个数组，使得两个子数组的和相等，而此题比上一题稍微难一点，因为上一题判断子数组的和比较简单，但是此题子数组的长度不是确定，只知道两个子数组的平均数是相同的。

蛋试我们还知道一个蕴含的条件就是大数组的平均数sum / n = 子数组1的平均数sum1 / i = 子数组2的平均数sum2 / (n - i)（这个证明比较简单，把sum1 + sum2 = sum代入大数组的平均数）

而大数组的平均数sum / n = 子数组1的平均数sum1 / i 这两个式子可以得出 sum * i / n = sum1（由于i和n - i具有对称性，所以我们只考虑i <= n - i）即i的取值范围为[1, n / 2]。所以我们可以利用sum * i % n == 0这个条件进行剪枝，去掉一些不可能的示例。蛋试满足此条件并不能判断A数组是否能够成功分配，比如[18,10,5,3]，sum = 36，当i = 1时sum * i % 4 == 0。所以需要进一步判断A数组中的选取i个元素的和能否达到sum1 = sum * i / n。

class Solution {
   
public:
    bool splitArraySameAverage(vector<int>& A) {
   
    	bool possible = false;
        int n = (int)A.size(), m = n / 2, sum = accumulate(A.begin(), A.end(), 0)