-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path3_Number_of_Ways_to_Split_Array.cpp
71 lines (56 loc) · 2.33 KB
/
3_Number_of_Ways_to_Split_Array.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
// 2270. Number of Ways to Split Array
// You are given a 0-indexed integer array nums of length n.
// nums contains a valid split at index i if the following are true:
// The sum of the first i + 1 elements is greater than or equal to the sum of the last n - i - 1 elements.
// There is at least one element to the right of i. That is, 0 <= i < n - 1.
// Return the number of valid splits in nums.
// Example 1:
// Input: nums = [10,4,-8,7]
// Output: 2
// Explanation:
// There are three ways of splitting nums into two non-empty parts:
// - Split nums at index 0. Then, the first part is [10], and its sum is 10. The second part is [4,-8,7], and its sum is 3. Since 10 >= 3, i = 0 is a valid split.
// - Split nums at index 1. Then, the first part is [10,4], and its sum is 14. The second part is [-8,7], and its sum is -1. Since 14 >= -1, i = 1 is a valid split.
// - Split nums at index 2. Then, the first part is [10,4,-8], and its sum is 6. The second part is [7], and its sum is 7. Since 6 < 7, i = 2 is not a valid split.
// Thus, the number of valid splits in nums is 2.
// Example 2:
// Input: nums = [2,3,1,0]
// Output: 2
// Explanation:
// There are two valid splits in nums:
// - Split nums at index 1. Then, the first part is [2,3], and its sum is 5. The second part is [1,0], and its sum is 1. Since 5 >= 1, i = 1 is a valid split.
// - Split nums at index 2. Then, the first part is [2,3,1], and its sum is 6. The second part is [0], and its sum is 0. Since 6 >= 0, i = 2 is a valid split.
// Constraints:
// 2 <= nums.length <= 105
// -105 <= nums[i] <= 105
class Solution
{
public:
int waysToSplitArray(vector<int> &nums)
{
long long sum = accumulate(nums.begin(), nums.end(), 0LL);
long long acc = 0LL;
int count = 0;
for (int i = 0; i < nums.size() - 1; i++)
{
acc += nums[i];
count += (2 * acc >= sum);
}
return count;
}
};
#include <vector>
#include <numeric>
using namespace std;
/*
Approach:
1. Calculate total sum of array using accumulate()
2. Keep track of running sum (acc) from left side
3. For each index i (except last):
- Add current element to acc
- Check if 2*acc >= sum (means left sum >= right sum)
- If true, increment count
4. Return total valid splits
Time: O(n) - One pass through array
Space: O(1) - Only using variables
*/