[BAEKJOON] 26531 Simple Sum

문제

You have hired someone to help you with inventory on the farm. Unfortunately, the new cowhand has very limited math skills, and they are having trouble summing two numbers. Write a program to determine if the cowhand is adding these numbers correctly.

입력

The first and only line of input contains a string of the form:

a + b = c

It is guaranteed that a, b, and c are single-digit integers. The input line will have exactly 9 characters, formatted exactly as shown, with a single space separating each number and arithmetic operator.

출력

Print on a single line, YES if the sum is correct; otherwise, print NO.

예제 입력 1 복사

1 + 2 = 3

예제 출력 1 복사

YES

예제 입력 2 복사

2 + 2 = 5

예제 출력 2 복사

NO

 

---

import sys

p = sys.stdin.readline().strip()

a = p.split()

if int(a[0]) + int(a[2]) == int(a[-1]):

    print("YES")

else:

    print("NO")