coidlity上的问题 之一 EQUI

codility上的问题都很简单，但是偶尔也有难一些的。想从慢慢做，发现可以把之前的全部问题都做一遍，反正题目也不多，索性做了。

题目：

找数组平衡点问题：

给定一个整数数组P，A[0..N - 1] ,平衡点定义为整数P,  满足

A[0] + A[1] +...+A[P - 1] = A[P + 1] + A[P + 2] + ... + A[N - 1]

注意不包含P元素。

输出任何一个平衡点，如果没有，则输出-1。

如果P == 0 等号左边是0，P == N - 1等号右边是0.

N是10^7的范围，每个数组元素是-2147483648..2147483647

要求复杂度: 时间复杂度O(N),空间复杂度O(1)

解答：这个题目比较容易。其实就是记录下全部数组元素的和sum（用long long)，然后试图枚举P, 顺便记录  left = A[0] +...+A[P - 1]

判断left == sum - left - A[P]即可。

// you can also use includes, for example:
// #include <algorithm>
int solution(const vector<int> &A) {
    // write your code here...
    int n = A.size(),i;
    long long sum = 0, left = 0;
    for (i = 0; i < n; ++i) {
        sum += A[i];
    }
    for (i = 0; i < n; ++i) {
        if (left == sum - A[i] - left) {
            return i;
        }
        left += A[i];
    }
    return -1;
}