[Python] SW Expert Academy #2005 파스칼의 삼각형

📚문제: #2005 파스칼의 삼각형(D2)

입력

1
4

출력

#1
1
1 1
1 2 1
1 3 3 1

전체 코드

# 2005. 파스칼의 삼각형

# 파스칼의 삼각형을 만드는 함수
def pascal_tri(array):
    
    # 초기화
    for i in range(N):
        
        array[i][0] = 1
        array[i][i] = 1
    
    # 연산
    for i in range(N):
        
        for j in range(N):
            
            if i-1 >= 0 and j-1 >= 0:
                
                array[i][j] = array[i-1][j-1] + array[i-1][j]
    
    # 출력
    for i in range(N):
        
        seq = []
        
        for j in range(i+1):
            
            seq.append(array[i][j])
        
        print(*seq)
            
            
# 테스트 케이스 T가 주어진다.
T = int(input())

# 테스트 케이스가 주어진다.
for p in range(1, T+1):
    
    # N이 주어진다.
    N = int(input())
    
    # N × N 2차원 리스트를 초기화 한다.
    grid = [[0] * N for _ in range(N)]
    
    print("#{}".format(p))
    
    pascal_tri(grid)