[코딩문제] 송전탑 문제
출처는 프로그래머스의 코딩테스트 입니다. 이 내용은 개인 공부를 위해 가져온 글이며 문제가 되면 삭제하겠습니다.
1. 문제 전문
설명
수평 직선에 높이가 서로 다른 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.
예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.
송신 탑(높이) | 수신 탑(높이) |
---|---|
5(4) | 4(7) |
4(7) | 2(9) |
3(5) | 2(9) |
2(9) | - |
1(6) | - |
맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.
제한 사항
- heights는 길이 2 이상 100 이하인 정수 배열입니다.
- 모든 탑의 높이는 1 이상 100 이하입니다.
- 신호를 수신하는 탑이 없으면 0으로 표시합니다.
입출력 예
heights | return |
---|---|
[6,9,5,7,4] | [0,0,2,2,4] |
[3,9,9,3,5,7,2] | [0,0,0,3,3,3,6] |
[1,5,3,6,7,6,5] | [0,0,2,0,0,5,6] |
2. 풀이 과정
처음에 for 문 안에서 for문을 돌리면 된다는 생각을 못하고 자꾸 조건과 While만을 떠올려서 많이 헤매였던 것 같다.
문제 핵심포인트
1) 기본값을 설정해주기
이 문제는 기본값이 존재할 수 있다. (자기보다 높은 송전탑이 없는 경우엔 0이니까)
이런 경우엔 기본값을 미리 세팅해놓고 그 위에 덮어쓰는 방식으로 풀면 좋은 것 같다.
기본값 설정 중 감명 깊었던 부분은
answer = []
answer = [0] * len(heights) < 요렇게 기본값을 설정해주는 것
=> answer = [0,0,0,0,0,] (heights배열 크기가 5라면)
--> 기본값이 존재할 수 있는 경우엔 기본값을 세팅해주고 기본값이 안 될 경우에만 기본값 위에 덮어쓰기
2) index를 가지고 놀기
배열안의 값을 비교할 때 배열안의 값 자체를 비교하면 나중에 결국 index를 불러들여야 하는 상황이 발생했다.
나중에 answer 리스트에 들어가는 값은 인덱스 값이 되므로 값을 비교할 땐 배열[인덱스] 방식으로 값을 비교하고 나중에 answer도 인덱스 값을 활용하면 쉽다.
--> 배열 내 값을 비교할 때 값자체가 아닌 인덱스로 비교하자
3) for문 안에서 for문 쓰기
배열내의 모든 값을 배열로 비교하려면 for문 안에 for문을 쓰는게 개꿀!
댓글
댓글 쓰기