[python] 백준 22940 선형 연립 방정식

https://www.acmicpc.net/problem/22940

 

문제 해결

• Reduced Row echelon form 
• 실수 부분 반올림 주의 (함부로 int 쓰지 말것)

 

CODE

import sys
input = sys.stdin.readline


n = int(input())
arr = [list(map(int, input().split())) for _ in range(n)]
for i in range(n):
    div = arr[i][i]
    for j in range(i, n + 1, 1):
        arr[i][j] /= div
    for j in range(n):
        if i==j:continue
        div = arr[j][i] # 계수가 1인 항의 몇배를 곱해서 없앨 수 있는지
        for k in range(n+1):
            arr[j][k] -= arr[i][k]*div
for i in range(n):
    print("{:.0f}".format(arr[i][n]),end=' ')