파스칼 삼각형

1. 문제

• 조합(Combination)을 손쉽게 구할 수 있는 파스칼 삼각형을 구현

2. 문제 설계

• n C 0 = 1, n C n = 1은 n개 중에서 0개, n개 중에서 n개를 뽑는 경우의 수이므로 1, 즉 2차원 배열(파스칼 삼각형)의 (i, 0), (i, i)는 모두 1로 채울 수 있음
• 그리고 나를 결정하기 위해서는 나의 왼쪽 위 대각선 값과 바로 위 값을 더해줌으로써 구할 수 있고, 최종적으로 n C r을 알기 위해서는 2차원 배열(파스칼 삼각형)의 (n, r)의 원소를 출력함으로써 값을 얻을 수 있음

3. 전체 코드

//MARK: - 파스칼 삼각형

//MARK: - Framework
import Foundation

//MARK: - Function
func solution() -> Void {
    //MARK: - Input
    guard let N: Int = Int(readLine() ?? "0") else { return }
    var pascalTriangle: [[Int]] = Array(repeating: Array(repeating: 0, count: N + 10), count: N + 10)
    
    //MARK: - Process
    pascalTriangle[0][0] = 1
    for i in 1...N {
        pascalTriangle[i][0] = 1    //n C 0 = 1
        pascalTriangle[i][i] = 1    //n C n = 1
        
        for j in 1..<i {
            pascalTriangle[i][j] = pascalTriangle[i - 1][j - 1] + pascalTriangle[i - 1][j]
        }
    }
    
    //MARK: - Output
    for i in 0...N {
        for j in 0..<N {
            if pascalTriangle[i][j] == 0 {
                continue
            }
            print("\(pascalTriangle[i][j]) ", terminator: "")
        }
        if pascalTriangle[i][N] == 0 {
            print()
        } else {
            print(pascalTriangle[i][N])
        }
    }
}
solution()

 

전체코드는 여기에서 확인할 수 있습니다.