[Java] BOJ.2493 - 탑

[3o14]

📣 Related Issue

제공되는 테스트 케이스 + 임의로 만든 테스트 케이스 모두 올바르게 나오는데
틀렸습니다 로 뜨는 코드입니다.
뭐가 문제일까요 😞

🌁 Background

백준 문제 풀이

👩‍💻 Contents

코드

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int N = Integer.parseInt(br.readLine());
		Integer[] arr = new Integer[N];
		Deque<Integer> stack = new ArrayDeque<>();
		List<Integer> list = new LinkedList<>();
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		// 1. 탑 입력받기
		for(int i=0; i<N; i++) {
			arr[i] = Integer.parseInt(st.nextToken()); 
		}
		
		// 2. 끝에서부터 탐색
		for(int i=N-1; i>=0; i--) {
			
			// 3. 스택이 비어있으면 push(addFirst)
			if(stack.isEmpty()) {
				stack.addFirst(arr[i]);
			} else {
				// 4. 현재 원소가 스택의 top보다 클 때: pop & list에 수신받은 탑 인덱스 add & 현재 원소 push
				if(stack.peek() < arr[i]) {
					while(!stack.isEmpty() && stack.peek() < arr[i]) { // 현재 원소보다 작은 값인 스택 요소를 전부 pop&list.add
						stack.poll();
						list.add(i+1);
					}
					stack.addFirst(arr[i]);
					
				// 5. 현재 원소가 스택의 top보다 작을 때: 스택에 push
				} else if(stack.peek() >= arr[i]) {
					stack.addFirst(arr[i]);
				}
			}
		}
		
		// 배열을 한바퀴 다 돌고도 스택에 남아있는 탑들 (자기자신보다 높은 탑을 발견하지 못해 수신받지 못한 탑들)
		while(!stack.isEmpty()) {
			int idx = Arrays.asList(arr).indexOf(stack.pollLast());
			list.add(arr.length-idx-1, 0); // 해당 원소 인덱스에 0 삽입
		}
		
		
		// 답 출력
		Collections.reverse(list);
		for(Integer i:list ) {
			sb.append(i).append(" ");
		}
		
		sb.deleteCharAt(sb.length() - 1); // 마지막 공백 제거
		System.out.println(sb);
	}
}

📱 Screenshot

📝 Review Note

백준에서 최종 제출한 코드는 pr작성을 위해 아래에서 참고한 코드로 제출한 것입니다!

📬 Reference

정답 코드 참고 링크

[minjae9610]

아직 풀지 못한 문제에 대한 질문은 이슈를 활용하셔도 좋을 것 같아요!

[minjae9610]

조만간 풀 문제 리스트에 있는 문제네요 :)
저도 풀고 한번 올려드릴께요

[chjcode]

제가 자바를 공부 중이라 잘 모르는 것일 수 도 있지만, 코드를 실행시켰을 때 5 5 5 5 5 와 같이 중복되는 값을 넣었을 때 오류가 났습니다. 또한 변수명을 보니 stack을 사용하려 계획 하신 것 같은데 addFirst, poll을 사용하신 것으로 보아 코드가 stack 자료구조를 활용하지 못한 것 같습니다.
좋은 문제 올려주신 덕분에 저도 재밌게 풀어보았습니다!