나의서재

<pre class="hljs"><code>public int[] solution(int[] enter, int[] exit){
 int totalWorker = enter.length;
 int[] answer = new int[totalWorker];

 // 나갈사람들
 Deque<Integer> exitWorkers = new LinkedList<>();
 for (int exitWorker : exit) {
 exitWorkers.offer(exitWorker);
 }

 // 만난 사람들을 검증하는 배열
 boolean[][] hasMeet = new boolean[totalWorker + 1][totalWorker + 1];
 for (int self = 0; self <= totalWorker; self++) {
 // 자기 자신과는 만날 수 없다.
 hasMeet[self][self] = true;
 }

 // 회의실에 있는 사람들
 Set<Integer> workersInMeetingRoom = new HashSet<>();
 while (!exitWorkers.isEmpty()) {
 for (int enterWorker : enter) {
 workersInMeetingRoom.add(enterWorker);

 for (int workerInMeetingRoom : workersInMeetingRoom) {
 if (!hasMeet[workerInMeetingRoom][enterWorker]) {
 answer[enterWorker - 1]++;
 answer[workerInMeetingRoom - 1]++;
 hasMeet[workerInMeetingRoom][enterWorker] = true;
 hasMeet[enterWorker][workerInMeetingRoom] = true;
 }
 }

 while (workersInMeetingRoom.contains(exitWorkers.peek())) {
 int exitWorker = exitWorkers.poll();
 workersInMeetingRoom.remove(exitWorker);
 }
 }
 }

 return answer;
}</code></pre> 저는 Set과 Queue(Deque)을 이용해서 풀어봤습니다. 컴파일 시 출력은 일치하지만 좋은 코드인지는 잘 모르겠습니다!

나의서재

게시글