目的:
給定一個(gè)整數(shù)類型的數(shù)組 nums,請編寫一個(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ù)組中不存在滿足此條件的中心索引。
說明:
nums 的長度范圍為 [0, 10000]。
任何一個(gè) nums[i] 將會是一個(gè)范圍在 [-1000, 1000]的整數(shù)。
解決思路:
利用sum、leftSum, rightSum分別存放總和,i左側(cè)和,i右側(cè)和,先算出一個(gè)總sum來是為了下面計(jì)算方便,然后移動一個(gè)從左到右的i,注意i在第一個(gè)元素時(shí)需要單獨(dú)考慮。分別算出i左邊還有右邊的值然后比對,如果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入門