[코딩 테스트] 조건 배열수 Python

주먹밥밥·2024년 3월 10일

문제💡

문자열에 따라 다음과 같이 두 수의 크기를 비교하려고 합니다.

두 수가 n과 m이라면
">", "=" : n >= m
"<", "=" : n <= m
">", "!" : n > m
"<", "!" : n < m
두 문자열 ineq와 eq가 주어집니다. ineq는 "<"와 ">"중 하나고, eq는 "="와 "!"중 하나입니다. 그리고 두 정수 n과 m이 주어질 때, n과 m이 ineq와 eq의 조건에 맞으면 1을 아니면 0을 return하도록 solution 함수를 완성해주세요.

입출력 예시

ineq	eq	n	m	result
"<"	    "="	20	50	1
">"	    "!"	41	78	0

입출력 예 설명
입출력 예

20 <= 50은 참이기 때문에 1을 return합니다.

입출력 예

41 > 78은 거짓이기 때문에 0을 return합니다.

문제 해석✏️

조건에 맞는 배열을 찾는 문제

  • if문을 통해 문자들이 같은 게 동시에 사용하고 있는지 확인(and 명령어를 통해 확인)
  • if문을 통해 숫자들이 비교가 제대로 되고 있는지 확인

if문을 쓰면 너무 길어질 것 같아서 쓰고 싶지 않았지만 생각 나는 게 이것밖에 없어 썼다.

문제

첫번쨰

def solution(ineq, eq, n, m):
    answer = 0
    if ineq =='>' and eq =='=':
        if n>=m:
            return 1

    elif ineq =='<' and eq =='=':
        if n<=m:
            return 1
        
    elif ineq =='>' and eq =='!':
        if n>m :
            return 1
        
    elif ineq =='<' and eq =='!':
        if n<m :
            return 1
        
    else:
        return 0

자꾸 ! 부분 test에서 오류가 났다.

문제점

  • if문으로 숫자 제대로 비교되고 있는지 확인 하지 않아도 됨
  • n,m이 정수인지 알려줘야 함

두번째

def solution(ineq, eq, n, m):
    if ineq == '>' and eq == '=':
        return int(n >= m)
    elif ineq == '<' and eq == '=':
        return int(n <= m)
    elif ineq == '>' and eq == '!':
        return int(n > m)
    elif ineq == '<' and eq == '!':
        return int(n < m)
    return 0
  • 바로 return 문을 해주면 비교수가 맞지 않으면 바로 return 0로 가주게 됨
  • 그 전에 썼던 숫자 비교 if문보단 확실한 정답인 것 같음.

다른 사람들의 풀이

def solution(ineq, eq, n, m):
    return int(eval(str(n)+ineq+eq.replace('!', '')+str(m)))

이렇게 쉽게 풀 수 있다니...너무 돌아갔다...

profile
코딩은 열심히 해야겠지...?

0개의 댓글