[코테/Python] 상자 안

목차 (OPEN)

• 📑 문제
• 💡 아이디어
• 👩🏻‍💻 전체 코드
• 📚 결론

 

 

 

 

📑 문제

• N * M 격자로 표현할 수 있는 바닥 위에 상자와 공이 놓여있다.
  • X: 상자의 위치
  • O: 공의 위치
  • . : 빈칸
• 상자는 직사각형 모양으로, 격자상에서 연속된 X들로 표현된다.
• 이때 공(O)이 상자의 테두리 안쪽에 있는지 확인해서
  상자 내부에 있다면 "Y"를, 그렇지 않다면 "N"을 출력해야 한다.

 

 

---

💡 아이디어

이거는 다들 어디선가 많이 봤던 문제일 것 같아서 아이디어라고 적기도 애매하지만 ..😅

그래도 간단히 적어보면 아래와 같다.

 

1. 상자의 테두리 좌표 구하기
   1) 왼쪽 위 꼭짓점의 좌표는 (r1, c1)
   2) 오른쪽 아래 꼭짓점의 좌표는 (r2, c2)
2. 공의 위치 확인하기
   공의 좌표 (br, bc)가 r1 < br < r2 그리고 c1 < bc < c2 범위 안에 들어가면 👉🏻 상자 안
   그렇지 않으면 👉🏻 상자 밖

 

 

이 그림처럼 결국 상자의 왼쪽 위 좌표와 오른쪽 아래 좌표만 구하면,

상자 내부에 존재하기 위한 점들의 x좌표와 y좌표의 범위를 바로 확인할 수 있다.

 

 

---

👩🏻‍💻 전체 코드

 

n, m = map(int, input().split())
s = [input().strip() for _ in range(n)]

r1 = n   # 왼쪽 위의 x좌표
r2 = -1  # 왼쪽 위의 y좌표
c1 = m   # 오른쪽 아래의 x좌표
c2 = -1  # 오른쪽 위의 y좌표

ball_r = -1 # 공의 x좌표
ball_c = -1 # 공의 y좌표

for i in range(n):
	for j in range(m):
		cur = s[i][j]
		if cur == 'X':
			if i < r1: r1 = i
			if i > r2: r2 = i
			if j < c1: c1 = j
			if j > c2: c2 = j
		elif cur == 'O':
			ball_r, ball_c = i, j

# 공이 상자 안에 있는지 판별
print('Y' if ((r1 < ball_r < r2) and (c1 < ball_c < c2)) else 'N')

 

• r1 = n 으로 초기화한 이유는,
  상자의 왼쪽 위 x좌표는 결국 가장 작은 행 번호여야 하기 때문에
  처음에는 충분히 큰 값(n)으로 두고, 더 작은 값이 나오면 갱신하기 위해서이다.
• r2 = -1 로 초기화한 이유도,
  상자의 오른쪽 아래 x좌표는 가장 큰 행 번호여야 하기 때문에
  처으멩는 충분히 작은 값(-1)으로 두고, 더 큰 값이 나오면 갱신하기 위해서이다.
• c1과 c2를 각각 위와 같이 초기화 한 이유도 동일하다.

 

 

---

📚 결론

교수님이 1차 코테는 엄청 어려워서 통과하는 사람의 거의 없을 거라고 겁주셨던 바람에

내 친구들 중에는 2차부터 참여하려고 1차는 들어오지 않았던 친구들도 있었는데,

이런 문제일 줄 알았으면 자기도 볼 걸 그랬다고 다들 엄청 아쉬워했다 ㅋㅋ

 

아무튼 앞으로도 이런 격자 문제를 풀 때,

전체 영역을 일일이 확인하기 보다는 경계 좌표만 기록해도 되는지를 떠올려서 확인해본다면, 훨씬 깔끔한 풀이가 가능할 것이다.