코테

[Codility] PassingCars C++

29도 맑음 2021. 12. 29. 16:42
반응형
int solution(vector<int> &A) {
    // write your code in C++14 (g++ 6.2.0)
    int east = 0;
    int west = 0;
    int result = 0;
    for(int i=0; i<A.size(); i++){
        if(A[i]){
            west++;
            result += east*west;
            if(result>1000000000)   return -1;
            west=0;
        }
        else{
            east++;
        }
    }
    return result;
}
반응형