java實(shí)現(xiàn)尋找數(shù)組中心索引的方法

目的：

給定一個(gè)整數(shù)類型的數(shù)組 nums，請(qǐng)編寫一個(gè)能夠返回?cái)?shù)組“中心索引”的方法。

數(shù)組中心索引定義：

數(shù)組中心索引的左側(cè)所有元素相加的和等于右側(cè)所有元素相加的和。如果數(shù)組不存在中心索引，那么我們應(yīng)該返回 -1。如果數(shù)組有多個(gè)中心索引，那么我們應(yīng)該返回最靠近左邊的那一個(gè)。

相關(guān)視頻教程推薦：java視頻

示例1：

輸入:  nums = [1, 7, 3, 6, 5, 6] 輸出: 3 解釋:  索引3 (nums[3] = 6) 的左側(cè)數(shù)之和(1 + 7 + 3 = 11)，與右側(cè)數(shù)之和(5 + 6 = 11)相等。 同時(shí), 3 也是第一個(gè)符合要求的中心索引。

示例2：

輸入:  nums = [1, 2, 3] 輸出: -1 解釋:  數(shù)組中不存在滿足此條件的中心索引。

說(shuō)明:

nums 的長(zhǎng)度范圍為 [0, 10000]。

任何一個(gè) nums[i] 將會(huì)是一個(gè)范圍在 [-1000, 1000]的整數(shù)。

解決思路：

利用sum、leftSum, rightSum分別存放總和，i左側(cè)和，i右側(cè)和，先算出一個(gè)總sum來(lái)是為了下面計(jì)算方便，然后移動(dòng)一個(gè)從左到右的i，注意i在第一個(gè)元素時(shí)需要單獨(dú)考慮。分別算出i左邊還有右邊的值然后比對(duì)，如果leftSum=rightSum，就返回i。

java代碼：

class Solution {     public int pivotIndex(int[] nums) {         if (nums == null || nums.length == 0) {             return -1;         }           int sum = 0;         int leftSum = 0;         int rightSum = 0;           for (int i = 0; i < nums.length; i++) {             sum += nums[i];         }           for (int i = 0; i < nums.length; i++) {             if (i == 0) {                 leftSum = 0;             } else {                 leftSum += nums[i - 1];             }             rightSum = sum - leftSum - nums[i];               if (leftSum == rightSum) {                 return i;             }         }           return -1;     } }

推薦教程：java入門