DEV Community

Prashant Mishra
Prashant Mishra

Posted on

No of ways to split Array

Problem

TC: O(n) for calculating prefix sum, and O(n) for iterating over the prefix sum for calculating valid splits

class Solution { public int waysToSplitArray(int[] nums) { long prefix[] = new long[nums.length]; long current = 0; for(int i=0;i<nums.length;i++){ current+=nums[i]; prefix[i] = current; } int count =0; for(int i =0;i<nums.length-1;i++){ long left = prefix[i]; long right = prefix[nums.length-1]; if(left>=right-left) count++; } return count; } } 
Enter fullscreen mode Exit fullscreen mode

Top comments (0)