给定的整数数组 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)