[1547] 공 / 파이썬 (feat. remove 함수)

 

 

 

 

✏️ 문제 링크

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

1547번: 공

---

❌ 실패....ㅠㅠ

리스트에서 숫자를 제거한다고 하니... pop을 써버렸다..ㅠ

일단 여기서 pop을 쓰면 안되는 이유 2가지

1. pop을 하면 제거된 숫자가 출력된다.

2. pop은 가장 마지막부터 꺼내진다.

---

❓ 정답은 ??

# 첫째 줄에 컵의 위치를 바꾼 횟수 M, M은 50보다 작거나 같은 자연수
# 둘째 줄부터 M개의 줄에는 컵의 위치를 바꾼 방법 X와 Y가 주어지며, 
# X번 컵과 Y번 컵의 위치를 서로 바꾸는 것을 의미한다. 
# X와 Y의 값은 3보다 작거나 같고, X와 Y가 같을 수도 있다.

m = int(input())

ball= 1
for _ in range(m):
    mix = list(map(int,input().split()))
    if ball in mix:
        mix.remove(ball)
        ball = mix[0]
    else: pass
    
print(ball)

---

💯 풀이 과정

1. 일단 공 위치 변수 설정을 해줘야한다.

2. 그리고 주어진 m 만큼 for 반복문을 시행

3. 위치 바꾼는 것은 리스트로 받는다.

4. 입력받은 두개 숫자 안에 공위치가 없으면 pass

5. in으로 있다고 하면, 바뀐다는 거니까 기존 공위치를 remove하고

6. 리스트 형태에서 뺀거니까, ball 변수에 int 형태로 갖고 있으려면 [0]를 써준다.

7. 그러면 끝!

---

😎 오늘의 한줄평

리스트 함수에 대해 잘 이해하자!