Advertisement
1988coder

724. Find Pivot Index

Jan 6th, 2019
116
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 0.77 KB | None | 0 0
  1. // LeetCode URL: https://leetcode.com/problems/find-pivot-index/
  2. /**
  3.  * Prefix Sum
  4.  *
  5.  * leftSum = rightSum (Excluding the pivot point)
  6.  *
  7.  * rightSum = TotalSum - leftSum - nums[pivotPoint]
  8.  *
  9.  * Time Complexity: O(N)
  10.  *
  11.  * Space Complexity: O(1)
  12.  *
  13.  * N = Length of the input nums array.
  14.  */
  15. class Solution {
  16.     public int pivotIndex(int[] nums) {
  17.         if (nums == null || nums.length == 0) {
  18.             return -1;
  19.         }
  20.  
  21.         int sum = 0;
  22.         int leftSum = 0;
  23.         for (int num : nums) {
  24.             sum += num;
  25.         }
  26.  
  27.         for (int i = 0; i < nums.length; i++) {
  28.             if (leftSum == sum - leftSum - nums[i]) {
  29.                 return i;
  30.             }
  31.             leftSum += nums[i];
  32.         }
  33.  
  34.         return -1;
  35.     }
  36. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement