백준 BAEKJOON 30642번 아이그루스와 화장실 [PYTHON/파이썬]

 

 

 

 

 

 

백준 BAEKJOON 30642번 아이그루스와 화장실 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅲ)

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

30642번: 아이그루스와 화장실

 

 

 

 

 

<풀이>

마스코트가 안뇽이면 홀수층, 인덕이면 짝수층에 배치해 주면 된다.

처음엔 현재 층수가 해당 층이 아니라면 K-1를 해서 정답을 유추하려고 했으나

현재 층수가 1층이라고 가정했을 경우에는 0층이 된다는 사실을 간과하였다.

그렇기에 if문을 통해 K+1층과 K-1층을 유도하였다.

 

 

 

 

 

 

<코드>

N = int(input())
mascot = input()
K = int(input())

if mascot == "annyong":
    if K % 2 == 1:
        print(K)
    else:
        print(K + 1 if K + 1 <= N else K - 1)
elif mascot == "induck":
    if K % 2 == 0:
        print(K)
    else:
        print(K + 1 if K + 1 <= N else K - 1)