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

题目

链接:leetcode链接


思路分析(前缀和)

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

left\] \[mid\] \[right

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

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

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

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

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

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

采用后缀和算法,

什么是后缀和算法呢?

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

具象到代码上就是

f----前缀和

g---后缀和

fi = fi-1 + numsi-1 注意,这里是numsi-1不是numsi,是因为求的是mid前,不包括mid

gi = gi+1 + numsi+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;
    }
相关推荐
zephyr0533 分钟前
从递归到迭代:二叉树非递归前中后序遍历详解
算法
Dr.kangder37 分钟前
嵌入式总线设备解析——TTE总线应用与实践
开发语言·网络·算法·嵌入式·多任务·同步机制
Hi李耶43 分钟前
【LeetCode】557.反转字符串中的单词 III
算法·leetcode·职场和发展
深蓝学院1 小时前
机器人学习算法五大体系详解:模仿、强化、多模态、持续学习……
算法·机器人
2401_858286111 小时前
OS82.【Linux】设计线程池
java·linux·运维·服务器·开发语言·算法·线程池
Epiphany.5562 小时前
找到所有好字符串(记忆化搜索+KMP)(里面含KMP模板,用来入门)
算法
坚持编程的菜鸟2 小时前
模拟实现strncpy
c语言·算法·模拟实现strncpy
深圳市快瞳科技有限公司4 小时前
个体识别、行为解读、健康管理:多模态宠物AI大模型的场景化落地
人工智能·算法·计算机视觉·大模型·多模态·宠物·宠物ai识别
liulilittle4 小时前
归一化:激活函数
人工智能·算法·机器学习·llm
Mem0rin5 小时前
前缀和:一维与二维
数据结构·算法