1. solution exception:索引超出了数组界限。_LintCode 题解|微软面试题:公平索引...

c053348e5db17c50d69df913f38bcc3d.png

现在给你两个长度均为N的整数数组 A 和 B。

当(A[0]+...A[K-1]),(A[K]+...+A[N-1]),(B[0]+...+B[K-1]) 和 (B[K]+...+B[N-1])四个和值大小相等时,称索引K是一个公平索引。也就是说,索引K 可以使得A, B 两个数组被分成两个非空数组,这四个子数组的和值相等。

例如,数组A = [4,-1,0,3],B = [-2,5,0,3],那么索引 K = 2是公平的,子数组的和相等:4+(-1) = 3; 0+3 = 3; -2 + 5 = 3 and 0 + 3 = 3。

现在请你计算公平索引的个数。

  • 2<=N<=100000
  • -1000000000<=a[i],b[i]<=1000000000 (0<=i<N)

在线评测地址:

LintCode 领扣​www.lintcode.com

样例 1:

输入: [4,-1,0,3] [-2,5,0,3]
输出: 2

样例 2:

输入: [2,-2,-3,3]
[0,0,4,-4]
输出: 1

样例 3:

输入: [4,-1,0,3] [-2,6,0,4]
输出: 0

样例 4:

输入: [1,4,2,-2,5] [7,-2,-2,2,5]
输出: 2

【题解】

先判断两个数组总和是否相等,若不相等,则直接返回0; 然后扫一遍数组,用pre_a和pre_b分别记录两个数组的前缀和,前缀和相等时ans++即可。 需要注意的是,数组中数的值在-1e9,1e9范围内,数组长度0,1e5,所以中间和会超出int范围,需要用long; 还有被分成的两个数组不能为空,所以前缀和p0和pn-1是不考虑的;

public class Solution {
    /**
     * @param A: an array of integers
     * @param B: an array of integers
     * @return: return a integer indicating the number of fair indexes.
     */
    public int CountIndexes(List<Integer> A, List<Integer> B) {
        int ans = 0;
        long sum_a = 0, sum_b = 0, pre_a = 0, pre_b = 0;
        for(int i = 0; i < A.size(); i++)
        {
            sum_a += A.get(i);
            sum_b += B.get(i);
        }
        if(sum_a != sum_b)return 0;
        for(int i = 0; i < A.size() - 1; i++)
        {
            pre_a += A.get(i);
            pre_b += B.get(i);
            if(pre_a==pre_b && pre_a == sum_a - pre_a)
                ans++;
        }
        return ans;
    }
}

更多语言代码参见:

九章算法 - 帮助更多中国人找到好工作,硅谷顶尖IT企业工程师实时在线授课为你传授面试技巧​www.jiuzhang.com
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值