数组平衡点问题

来源:互联网 发布:js冒泡排序 从小到大 编辑:程序博客网 时间:2024/05/01 00:55

题目:

找数组平衡点问题:

给定一个整数数组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]即可。

[cpp] view plaincopy
  1. // you can also use includes, for example:  
  2. // #include <algorithm>  
  3. int solution(const vector<int> &A) {  
  4.     // write your code here...  
  5.     int n = A.size(),i;  
  6.     long long sum = 0, left = 0;  
  7.     for (i = 0; i < n; ++i) {  
  8.         sum += A[i];  
  9.     }  
  10.     for (i = 0; i < n; ++i) {  
  11.         if (left == sum - A[i] - left) {  
  12.             return i;  
  13.         }  
  14.         left += A[i];  
  15.     }  
  16.     return -1;  
  17. }  
0 0
原创粉丝点击