C - Takahashi's Information

C - Takahashi's Information

O(1) 問題文から下記の要素が推測できる。 引き算をして各要素が等しいか判定。

a_1 + b_1,  a_1 + b_2, a_1 + b_3 \\
a_2 + b_1,  a_2 + b_2, a_2 + b_3 \\
a_3 + b_1,  a_3 + b_2, a_3 + b_3 \\

import math
import heapq
import itertools
from functools import reduce
 
# main
def main():
    C = []

    for _ in range(0, 3):
        row = list(map(int, input().split()))
        C.append(row)

    ok = True
    if C[0][0] - C[0][1] != C[1][0] - C[1][1] or C[1][0] - C[1][1] != C[2][0] - C[2][1]:
        ok = False
    if C[0][1] - C[0][2] != C[1][1] - C[1][2] or C[1][1] - C[1][2] != C[2][1] - C[2][2]:
        ok = False
    
    if ok:
        print("Yes")
    else:
        print("No")

# エントリポイント
if __name__ == '__main__':
    main()