算法:724.寻找数组的中心下标

题目

链接:leetcode链接


思路分析(前缀和)

根据题意,我们可以将数组看成三个部分

left\] \[mid\] \[right

我们只需要[left]区间内的元素和等于[right]区间的元素和即可,此时mid就是中心下标

那么我们可以借助前缀和思想,将[left]区间和[right]区间的元素和求出来,就可以很好的方便比较

注意,前缀和算法不仅仅是一种算法,更是一种思想。

看到这里,读者们可能会问,通过前缀和算法为什么可以求出[right]区间的和呢?

当然,用整个数组的和减去[left]区间和mid元素可以得到答案,

但是,这里介绍一种更好的办法

采用后缀和算法,

什么是后缀和算法呢?

顾名思义,将前缀和反过来,我们从数组末尾向前求和就成了后缀和

具象到代码上就是

f----前缀和

g---后缀和

f[i] = f[i-1] + nums[i-1] 注意,这里是nums[i-1]不是nums[i],是因为求的是mid前,不包括mid

g[i] = g[i+1] + nums[i+1]

最后处理好前缀和数组和后缀和数组之后,只需要遍历比较前缀和数组和后缀和数组即可。


代码

C++ 复制代码
int pivotIndex(vector<int>& nums) {
        int n = nums.size();
        vector<int> f(n);
        vector<int> g(n);

        for(int i = 1;i< n;++i)
            f[i] = f[i-1] + nums[i-1];
        for(int i = n-2;i>=0;--i)
            g[i] = g[i+1] + nums[i+1];

        for(int i = 0;i < n;++i)
        {
            if(f[i] == g[i])
            return i;
        }

        return -1;
    }
相关推荐
Yingye Zhu(HPXXZYY)38 分钟前
ICPC 2023 Nanjing R L 题 Elevator
算法
程序员Xu4 小时前
【LeetCode热题100道笔记】二叉树的右视图
笔记·算法·leetcode
笑脸惹桃花4 小时前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
阿维的博客日记5 小时前
LeetCode 48 - 旋转图像算法详解(全网最优雅的Java算法
算法·leetcode
GEO_YScsn5 小时前
Rust 的生命周期与借用检查:安全性深度保障的基石
网络·算法
程序员Xu5 小时前
【LeetCode热题100道笔记】二叉搜索树中第 K 小的元素
笔记·算法·leetcode
THMAIL6 小时前
机器学习从入门到精通 - 数据预处理实战秘籍:清洗、转换与特征工程入门
人工智能·python·算法·机器学习·数据挖掘·逻辑回归
Kevinhbr6 小时前
CSP-J/S IS COMING
数据结构·c++·算法
蕓晨7 小时前
set的插入和pair的用法
c++·算法
THMAIL7 小时前
深度学习从入门到精通 - AutoML与神经网络搜索(NAS):自动化模型设计未来
人工智能·python·深度学习·神经网络·算法·机器学习·逻辑回归