Time: 1 ms (100.00%), Space: 41.9 MB (100.00%) - LeetHub

py3-coder · py3-coder · commit 89276ad38d07 · 2024-11-18T11:38:45.000+05:30
diff --git a/3354-make-array-elements-equal-to-zero/3354-make-array-elements-equal-to-zero.java b/3354-make-array-elements-equal-to-zero/3354-make-array-elements-equal-to-zero.java
@@ -0,0 +1,26 @@
+class Solution {
+    public int countValidSelections(int[] nums) {
+        int cnt =0;
+        int n = nums.length;
+        for(int i=0;i<n;i++){
+            if(nums[i]==0){
+                int lfs = sum(0,i,nums);
+                int rfs = sum(i+1,n,nums);
+                if(lfs == rfs ){
+                    cnt+=2;
+                }else if(Math.abs(lfs-rfs)==1){
+                    cnt+=1;
+                }
+            }
+        }
+        return cnt ;
+    }
+    
+    public int sum(int start , int end , int[] nums){
+        int ans =0; 
+        for(int i= start ; i<end;i++){
+            ans +=nums[i];
+        }
+        return ans ;
+    }
+}
