寻找数组的中心下标 ---- 前缀和

题目接

题目:

分析:

  • 这道题是求某个区间内数组的和, 所以可以用前缀和算法
  • 假设i位置是中心位置, 我们要知道[0,i - 1]的和 和 [i + 1,n - 1]的和 看是否相等
  • 先求[0,i - 1]的和 用一个前缀数组f
    此时f[i] 表示: 前i - 1个数的和, 那么f[i - 1] 就表示前i - 2的和, 所以让f[0] = 0, i从1开始, 从前往后遍历
    f[i] = f[i - 1] + nums[i - 1]
  • 求[i + 1,n - 1]的和 用一个后缀数组g
    此时g[i] 表示: 后i + 1个数的和, 那么g[i + 1] 就表示后i + 2的和, 所以让g[n - 1] = 0, i从n-2开始, 从后往前遍历
    g[i] = g[i + 1] + nums[i + 1]
  • 从头遍历数组, 判断f[i] 是否等于 g[i]

代码:

java 复制代码
class Solution {
    public int pivotIndex(int[] nums) {
        int n = nums.length;
        int[] f = new int[n];
        int[] g = new int[n];
        for (int j = 1; j < n; j++) {
            f[j] = f[j - 1] + nums[j -1];
        }
        for (int j = n - 2; j >= 0; j--) {
            g[j] = g[j + 1] + nums[j + 1];
        }
        for (int i = 0; i < n; i++) {
            if(f[i] == g[i]){
                return i;
            }
        }
        return -1;
    }
}
相关推荐
ysa05103017 分钟前
竞赛常用加速技巧#模板
c++·笔记·算法
7 9719 分钟前
C语言基础知识--文件的顺序读写与随机读写
java·数据结构·算法
2401_8410039820 分钟前
Kubernetes 资源管理全解析
算法·贪心算法
☆璇1 小时前
【数据结构】排序
c语言·开发语言·数据结构·算法·排序算法
艾莉丝努力练剑4 小时前
【LeetCode&数据结构】单链表的应用——反转链表问题、链表的中间节点问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
_殊途6 小时前
《Java HashMap底层原理全解析(源码+性能+面试)》
java·数据结构·算法
珊瑚里的鱼9 小时前
LeetCode 692题解 | 前K个高频单词
开发语言·c++·算法·leetcode·职场和发展·学习方法
秋说10 小时前
【PTA数据结构 | C语言版】顺序队列的3个操作
c语言·数据结构·算法
lifallen11 小时前
Kafka 时间轮深度解析:如何O(1)处理定时任务
java·数据结构·分布式·后端·算法·kafka
liupenglove11 小时前
自动驾驶数据仓库:时间片合并算法。
大数据·数据仓库·算法·elasticsearch·自动驾驶