题目:
找数组平衡点问题:
给定一个整数数组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;
- }