프로그래머스
13. 프로그래머스_탑
UDUD
2019. 3. 25. 15:33
반응형
Using Language : Java
Using Tool : IntelliJ
- 프로그래머스 문제 소개
< 탑 >
문제 설명
수평 직선에 탑 N대를 세웠습니다.
모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다.
발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다.
또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.
예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다.
그러면, 탑은 다음과 같이 신호를 주고받습니다.
높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고,
높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이,
높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다.
높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는
어떤 탑에서도 수신할 수 없습니다.
제한 조건
- 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]
의사코드
- 정답 배열의 첫번째는 무조건 0 을 넣어준다.
- 기준 탑을 정하기 위한 for문과, 비교하는 탑을 정하기 위한 for문을 작성
- 비교하는 탑이 기준 탑보다 클 경우 정답 배열에 비교하는 탑의 위치를 넣어준다. 이 경우 탑의 위치는 1부터 시작하므로, 0부터 시작하는 배열 인덱스를 그냥 넣지 말고 1을 더해서 값을 넣어준다.
<Solution16.java>
class Solution16 {
public int[] solution(int[] heights) {
int[] answer = new int[heights.length];
answer[0] = 0;
int len = heights.length;
for (int i = 1; i < len; i++) {
for (int j = 0; j < i; j++) {
if (heights[i] < heights[j]) {
answer[i] = j + 1;
}
}
}
return answer;
}
}
<Solution16Test.java>
import static org.junit.Assert.assertArrayEquals;
import org.junit.Test;
public class Solution16Test {
@Test
public void 결과() {
Solution16 solution = new Solution16();
int[]heights = {6,9,5,7,4};
int[]heights2= {3,9,9,3,5,7,2};
int[]heights3= {1,5,3,6,7,6,5};
int[]answer= {0,0,2,2,4};
int[]answer2= {0,0,0,3,3,3,6};
int[]answer3= {0,0,2,0,0,5,6};
assertArrayEquals(answer,solution.solution(heights));
assertArrayEquals(answer2,solution.solution(heights2));
assertArrayEquals(answer3,solution.solution(heights3));
}
}
반응형