Medium
Number of Ways to Split Array — C++
Full explanation · Time O(n) · Space O(1)
// Time: O(n)
// Space: O(1)
// prefix sum
class Solution {
public:
int waysToSplitArray(vector<int>& nums) {
int64_t total = accumulate(cbegin(nums), cend(nums), 0ll), curr = 0;
int result = 0;
for (int i = 0; i + 1 < size(nums); ++i) {
curr += nums[i];
result += static_cast<int>(curr >= (total - curr));
}
return result;
}
};