codility Lesson3 - TapeEquilibrium

코드

시간복잡도: O(N)

// you can use includes, for example:
// #include <algorithm>

// you can write to stdout for debugging purposes, e.g.
// cout << "this is a debug message" << endl;
#include <cmath>

int solution(vector<int> &A) {
    // write your code in C++14 (g++ 6.2.0)

    long long sumA = 0;
    for(auto a : A) {
        sumA += a;
    }

    long long min = 999999;
    long long sumP = 0;
    for(int i=0;i<A.size()-1;i++) {
        sumA -= A[i];
        sumP += A[i];
        min = min > abs(sumA-sumP) ? abs(sumA-sumP) : min;
    }

    return min;
}

풀이 및 소감

먼저 A를 돌면서 합을 구하고 다음 for문에서 하나씩 빼주면서 새로운 변수에서 다시 합을 구하고 두개의 차이를 비교하면서 해결했다.