[Codility] PassingCars 70→100

문제출처

 https://app.codility.com/programmers/lessons/5-prefix_sums/passing_cars/

 

문제요약

 0은 ->으로 가는차, 1은 <-으로 가는차

 0의 각각 차들이 마주치는 차들의 합을 구하는 문제

코드 

70점  정답100%, 퍼포먼스 몇가지 time out

https://app.codility.com/demo/results/training9SAGSM-FTW/

1방향의 카운트를 모두 센다음, 반복문을 돌면서 

0일때는 1의 카운트값을 넣고

1일대는 1의 카운트값을 담은 값을 감소시킨다. (다음차로 넘어 갔다는 의미)

class Solution {
    public int solution(int[] A) {
        
        int result = 0;
        int oneCount = 0;
 
        for(int i=0; i<A.length; i++){
            if(A[i] == 1) oneCount++;
        }
 
        for(int j=0; j<A.length; j++){
            if(A[j] == 0){
                result += oneCount;
            }else{
                oneCount--;
            }
        }
 
        return result;
    }
}

100점

 https://app.codility.com/demo/results/trainingT8RMZP-B8C/

예외처리를 추가하였음 17행

class Solution {
    public int solution(int[] A) {
        int result = 0;
        int oneCount = 0;
 
        for(int i=0; i<A.length; i++){
            if(A[i] == 1) oneCount++;
        }
 
        for(int j=0; j<A.length;j++){
            if(A[j] == 0){
                result += oneCount;
            }else{
                oneCount--;
            }
 
            if(result>1000000000){
                result = -1;
                break;
            }
        }
 
        return result;
    }
}