[프로그래머스] 3. OX 퀴즈 - Python

프로그래머스 : OX 퀴즈

 

• 프로그래밍 언어 : Python

 

문제집 링크 : http://aladin.kr/p/szPra

내일은 코딩테스트 with 파이썬

 

프로그래머스 URL :  

[도서실습] 내일은 코딩테스트 with 파이썬(자료구조와 알고리즘의 기초부터 실전까지)

---

■ 작성 코드

def solution(quiz):
    answer = []  
    for p in quiz:
        p_a = p.split()    
        
        if p_a[1]=='+':
            if int(p_a[0])+int(p_a[2]) == int(p_a[-1]):
                ans = "O"
                answer.append(ans)
            else:
                ans = "X"
                answer.append(ans)
        elif p_a[1]=='-':
            if int(p_a[0])-int(p_a[2]) == int(p_a[-1]):
                ans = "O"
                answer.append(ans)
            else:
                ans = "X"
                answer.append(ans)    
                
    return answer

 

1. quiz의 요소를 for문에 따라서 split()

2. p_a의 0번째 요소는 숫자, p_a의 2번째 요소도 숫자 그리고 마지막 요소도 숫자라는 것을 이용

3. p_a의 1번째 요소가 연산자라는 것을 이용하여 if 문 작성