724. Find Pivot Index - Easy

前往題目

想法

• 算出每個點的左邊sum，以及右邊sum，最後再看是否有一樣的即可

思路

同想法

Code

class Solution {
    public int pivotIndex(int[] nums) {
        int[] sumLeft = new int[nums.length];
        int[] sumRight = new int[nums.length];

        sumLeft[0] = 0;
        sumRight[sumRight.length - 1] = 0;

        // 計算左邊sum
        for (int i = 1; i < nums.length; ++i) {
            sumLeft[i] += sumLeft[i - 1] + nums[i - 1];
        }
        // 計算右邊sum
        for (int i = nums.length - 2; i >= 0; --i) {
            sumRight[i] += sumRight[i + 1] + nums[i + 1];
        }
        // 判斷是否有相同的元素
        for (int i = 0; i < nums.length; ++i) {
            if (sumLeft[i] == sumRight[i]) return i;
        }

        return -1;
    }
}