724. Find Pivot Index

题目连接：

https://leetcode-cn.com/problems/find-pivot-index/

题目大意：

找一个点，左边和右边相等。

具体思路：

前缀和+后缀和，线性遍历一遍即可。一定注意存在空数组的情况。

AC代码：

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int sum_pre[10000+10];
        int sum_suf[10000+10];
        memset(sum_pre,0,sizeof(sum_pre));
        memset(sum_suf,0,sizeof(sum_suf));
        int len = nums.size();
        int ans = -1 ;
        if(len){
        sum_pre[0] = nums[0] , sum_suf[len-1] = nums[len-1];
         
        for ( int i = 1 ; i< len ;i++ ){
           sum_pre[i] = sum_pre[i-1] + nums[i];
        }
        for( int i = len-2 ;i >=0 ;i--){
           sum_suf[i] = sum_suf[i+1] + nums[i];
        }
        for(int i = 0 ;i <len ;i++){
         if( i == 0 ){
           if(sum_suf[i+1] == 0 ){
               ans=i;
               break;
           }
         }
        else if(i == len-1){
        if(i - 1 >= 0 && sum_pre[i-1] == 0){
        ans=i;
            break;
        }
        }
        else{
            if(i + 1< len && i - 1 >= 0 && sum_pre[i-1] == sum_suf[i+1] ){
            ans=i;
                break;
            }
        }
        }
        }
        return ans;
    }
};