[BOJ] 16395. 파스칼의 삼각형(🥈, DP)

📝 문제

풀이

n이 1 or 2면 k의 값에 상관없이 답은 1이 된다
그 외에는 양쪽에 1을 붙이고 dp로 이전의 값들을 더해나가면서 구하면 된다

# 파스칼의 삼각형

def sol(n, k):
    if n == 1 or n == 2:
        return 1
    
    dp = [1, 1]
    i = 2
    while i < n:
        new_dp = [1]
        for j in range(len(dp)-1):
            new_dp.append(dp[j]+dp[j+1])
        new_dp.append(1)
        i += 1
        dp = new_dp[:]
    
    return dp[k-1]

n, k = map(int, input().split())

print(sol(n, k))