https://leetcode-cn.com/problems/find-pivot-index/
思路:枚举中间位置
i
i
i,看
[
0
,
i
)
[0,i)
[0,i)和
(
i
,
n
)
(i,n)
(i,n)元素之和是否相等即可,通过前缀和降低复杂度。
class Solution {
public:
int pivotIndex(vector<int>& nums) {
int sum=0,lsum=0,n=nums.size();
for(int i=0;i<n;i++)
sum+=nums[i];
for(int i=0;i<n;i++)
{
sum-=nums[i];
if(sum==lsum)
return i;
lsum+=nums[i];
}
return -1;
}
};