面试题12:左右两边子数组的和相等(Java版)

题目:输入一个整数数组,如果一个数字左边的子数组的数字之和等于右边的子数组的数字之和,那么返回该数字的下标。如果存在多个这样的数字,则返回最左边一个数字的下标。如果不存在这样的数字,则返回-1。例如,在数组[1,7,3,6,2,9]中,下标为3的数字(值为6)的左边3个数字1、7、3的和与右边两个数字2和9的和相等,都是11,因此正确的输出值是3。

// 利用右边的数字之和等于总的数字之和减去左边
// 的数字之和与分界点的那个数字这个规律来比较两
// 边的数是否相同
public static int pivotIndex(int[] nums) {
    // total为所有数的和
    int total = 0;
    for (int num : nums) {
        total += num;
    }
    // sum为从最初的位置到当前位置数字之和
    int sum = 0;

    for (int i = 0; i < nums.length; i++) {
        sum += nums[i];
        // sum - nums[i] = 当前位置左边的所有数字之和
        // total - sum = 当前位置右边数字之和
        if (sum - nums[i] == total - sum) {
            return i;
        }
    }
    return -1;
}

你可能感兴趣的:(剑指office(Java版),leetcode,算法,java)