題目描述
給定一個(gè)整數(shù)類型的數(shù)組 nums,請(qǐng)編寫一個(gè)能夠返回?cái)?shù)組“中心索引”的方法。
我們是這樣定義數(shù)組中心索引的:數(shù)組中心索引的左側(cè)所有元素相加的和等于右側(cè)所有元素相加的和。
如果數(shù)組不存在中心索引,那么我們應(yīng)該返回 -1。如果數(shù)組有多個(gè)中心索引,那么我們應(yīng)該返回最靠近左邊的那一個(gè)。
示例 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的長(zhǎng)度范圍為[0, 10000]。 - 任何一個(gè)
nums[i]將會(huì)是一個(gè)范圍在[-1000, 1000]的整數(shù)。
第一次提交
class Solution {
public int pivotIndex(int[] nums)
{
int p=0;
for (p = 0; p<=nums.length-1; p++)
{
int sum_left = 0;
int sum_right = 0;
for (int i = 0;i<=p-1;i++)
sum_left += nums[i];
for (int j = p+1;j<=nums.length-1; j++)
sum_right += nums[j];
if (sum_left == sum_right)
return p;
}
return -1;
}
}
這個(gè)解答的思路就是暴力破解,遍歷數(shù)組中每一個(gè)元素,把這個(gè)元素左邊所有的元素和求出來,再把這個(gè)元素右邊所有的元素和求出來,比較是否相等??上攵@個(gè)速度是很慢的。
改進(jìn)
class Solution {
public int pivotIndex(int[] nums)
{
int sum = 0;
for (int num :
nums) {
sum += num;
}
int sum_left = 0;
for (int p = 0; p<nums.length; p++)
{
if (p!=0)
{
sum_left += nums[p-1];
}
int sum_right = sum - nums[p] - sum_left;
if (sum_left == sum_right)
return p;
}
return -1;
}
}
這個(gè)是看了別人的解答之后做出的修改。首先把數(shù)組每一個(gè)元素累加起來,得到一個(gè)和,然后依然是遍歷數(shù)組中的每一個(gè)元素,每遍歷一個(gè)元素,就使sum_left加上當(dāng)前元素的前一個(gè)元素,同時(shí)右邊元素和等于元素總和減去當(dāng)前元素再減去左邊元素和,這樣就大大減少了做加法的次數(shù)。比較每一次左邊元素和和右邊元素和,若相等就返回索引值。
感想
說出來不怕大家笑話,做了有四五個(gè)題了吧,都是簡(jiǎn)單等級(jí)的,這還是第二個(gè)獨(dú)立做出來的,還是用的暴力破解。不過,跟前兩天不看答案不會(huì)下手相比還是有進(jìn)步的。接下來的題目,先想出暴力破解的方案,但是不要急著提交,可能提交了也達(dá)不到時(shí)間要求,要多思考一下能不能剔除一些情況,簡(jiǎn)化計(jì)算量。