Codility - Passing Cars

Passing Cars

문제

N의 길이로 이루어진 배열 A는 0과 1로 이루어져 있는데, 0과 1은 각각 다음과 같은 의미를 가지고 있다.

• 0은 차가 동쪽으로 간다
• 1은 차가 서쪽으로 간다

이 때 동쪽으로 간 차와 서쪽으로 간 차를 짝지을 수 있는 개수를 구하라. 단 먼저 동쪽으로 간차와 그 이후에 서쪽으로 간 차만 짝 지을 수 있다.

A배열이 아래와 같이 주어져 있다면
A[0] = 0
A[1] = 1
A[2] = 0
A[3] = 1
A[4] = 1

짝 지을 수 있는 경우의 수는 (0, 1), (0, 3), (0, 4), (2, 3), (2, 4).

5가지다.

단 짝의 개수가 1,000,000,000개를 넘어가면 그냥 -1을 리턴한다.

풀이

// you can write to stdout for debugging purposes, e.g.
// console.log('this is a debug message');

function solution(A) {
  // 동쪽으로 간차의 개수를 센다
  let east = 0
  // 결과
  let passing = 0

  for (let i of A) {
    // 동쪽으로 간 차를 센다.
    if (i === 0) {
      east += 1
    } else {
      // 서쪽으로 간 차가 나타난다면, 현재 동쪽으로 간 차 개수만큼 더한다.
      // 현재 동쪽으로 간 차 개수만큼 짝이 될 수 있기 때문이다.
      passing += east
    }
  }

  if (passing > 1000000000) {
    return -1
  }

  return passing
}

https://app.codility.com/demo/results/trainingXFPYT4-R3D/