顺序表vector--------练习题7题解

题目

解法

class Solution {
public:
    int findMiddleIndex(vector<int>& nums) {
        int l, r, mid = 0, suml = 0, sumr = 0;
        while (mid < nums.size()) {
            l = mid - 1;
            r = mid + 1;
            suml = 0, sumr = 0;
            while (l >= 0) {
                suml += nums[l];
                l--;
            }
            while (r < nums.size()) {
                sumr += nums[r];
                r++;
            }
            if (suml == sumr) {
                return mid;
            }
            mid++;
        }
        return -1;
    }
};

反思

1.考察知识点：顺序表vector，双指针

2.踩的坑：每次循环的总和numl,numr没有放在循环里面进行刷新；

3。可以优化的地方:现在这个解法的时间复杂度是O(n^2),空间复杂度是O(1).对时间进一步优化的话，可以采用前缀和+总和公式的方法：

class Solution {
public:
    int findMiddleIndex(vector<int>& nums) {
        int total = 0;
        // 1. 先算数组总和（仅遍历1次，O(n)）
        for (int i=0;i<nums.size();i++) total += nums[i];
        
        int suml = 0;
        // 2. 遍历每个mid，用公式算右和（O(n)）
        for (int mid = 0; mid < nums.size(); mid++) {
            // 右和 = 总和 - 左和 - 当前mid的值
            int sumr = total - suml - nums[mid];
            if (suml == sumr) return mid;
            // 左和累加当前元素，为下一个mid做准备
            suml += nums[mid];
        }
        return -1;
    }
};

从这个思路中我可以学习到，左和+mid+右和就等于总和，所以无需单独计算，只需要一个for循环来判断大小就好了。