724. 寻找数组的中心下标

代码:

class Solution {
    public int pivotIndex(int[] nums) {
        int last=0;    //放前一个下标的左右差值
        int now=0;     //放当前下标的左右差值

        // 0 位于边界,先算出 0 下标左右的差值
        for(int i=1;i<nums.length;i++){
            last+=nums[i];
        }

        //判断是否 0 下标就已经是中心下标
        if(last==0){
            return 0;
        }

        //获取各个下标左右两边的差值
        for(int i=1;i<nums.length;i++){
            now=last-nums[i]-nums[i-1];
            if(now==0){
                return i;
            }
            last=now;
        }

        return -1;
    }
}

题解:

        本题的意思很简单,我们需要找到一个中心下标,中心下标左边和右边的数据和是相等的,在遇到一个题时,我们先想一下它的暴力解法,因为优化解法通常是由暴力解法演化而来的

        本题的暴力解法也很简单,我们可以去求每个下标左右两边的数据和是否相等,每判断一次下标就需要遍历一次数组,而我们最多要判断 n 次,所以时间复杂度为 O(n),这是一个比较大的复杂度了,我们要想一个优化解法

        要判断左右两边的数据是否相同,我们可以判断左右两边的数据和之差是否为 0 ,如果为 0 ,就说明左右两边的数据和相同,我们可以利用数组 dp 来存储对应下标左右两边的数据和之差,如 dp[ i ] ,就代表 i 下标左右两边的数据和之差,当 dp[ i ] = 0 时就代表  i 下标左右两边的数据和相同,i 下标就为中心下标

        现在我们要探究的就是如何填充 dp 数组,我画了下面的图方便理解

        如上图,我们要获取 dp[ i ] 的值就需要 a - b, a = c - nums[ i ],b = d + nums[ i-1 ],所以 a - b =

c - nums[ i ] - d - nums[ i-1 ] ,而 c - d = dp[ i-1 ],所以 dp[ i ] = dp[ i-1 ] - nums[ i ]  - nums[ i-1 ],

我们通过上面的式子从左到右便可以填充完 dp 数组

        要注意,我们首先要计算 dp[ 0 ] 的值,因为当 i =0 时,在该式子中会出现 -1 下标,这是越界的,所以我们先要计算出 0 下标左右两边数据和的差值,再从左到右填充 dp 数组,当 填充到 dp 数组中的值为 0 时,就找到了中心下标

        实际上,我们在填充 dp 数组时,只需要 dp 前一个下标的值,并且只要 dp 数组中的值出现 0 就不需要遍历了,所以我们直接可以用两个变量,一个记录当前下标的两边数据和的差值,一个记录前一个下标两边数据和的差值,就能代替 dp 数组

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小林想被监督学习

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值