[백준] 자바 7785 회사에 있는 사람

문제

상근이는 세계적인 소프트웨어 회사 기글에서 일한다. 이 회사의 가장 큰 특징은 자유로운 출퇴근 시간이다. 따라서, 직원들은 반드시 9시부터 6시까지 회사에 있지 않아도 된다.

각 직원은 자기가 원할 때 출근할 수 있고, 아무 때나 퇴근할 수 있다.

상근이는 모든 사람의 출입카드 시스템의 로그를 가지고 있다. 이 로그는 어떤 사람이 회사에 들어왔는지, 나갔는지가 기록되어 있다. 로그가 주어졌을 때, 현재 회사에 있는 모든 사람을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 로그에 기록된 출입 기록의 수 n이 주어진다. (2 ≤ n ≤ 106) 다음 n개의 줄에는 출입 기록이 순서대로 주어지며, 각 사람의 이름이 주어지고 "enter"나 "leave"가 주어진다. "enter"인 경우는 출근, "leave"인 경우는 퇴근이다.

회사에는 동명이인이 없으며, 대소문자가 다른 경우에는 다른 이름이다. 사람들의 이름은 알파벳 대소문자로 구성된 5글자 이하의 문자열이다.

출력

현재 회사에 있는 사람의 이름을 사전 순의 역순으로 한 줄에 한 명씩 출력한다.

---

테스트 케이스

 

초록색 입력/ 검은색 출력

 

---

접근

원하는 수만큼 입력을 받아서, 그 수 만큼 반복문을 돌며 enter와 leave를 구분하여 현재 enter 돼있는 사람만 추려서 역순으로 출력해주는 문제입니다.

중복을 제거하기위해 hashset을 이용하였고, stringTokenizer로 문자열을 잘라서 enter와 leave를 구별하여 enter라면 해쉬 셋에 이름을 넣어주고 leave라면 해쉬 셋에서 제거하는 방식으로 구현하였습니다.

 

해쉬 셋은 자동정렬 기능이 없어서 arraylist로 옮겨준 후, 자동 정렬하여 출력하였습니다.

---

코드

import java.awt.desktop.SystemEventListener;
import java.io.*;
import java.math.*;
import java.util.*;

public class Main {

	/*
 	7785 problem 회사에 있는 사람
	*/
	public static void main(String[] args) throws NumberFormatException, IOException {
		// TODO Auto-generated method stub
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		TreeSet<String> set =new TreeSet<>();
		ArrayList<String> arr = new ArrayList<>();
		
		int peopleNum = Integer.parseInt(br.readLine());
		
		for(int i=0; i<peopleNum; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			String name = st.nextToken();
			String com = st.nextToken();
			
			//enter면 해쉬셋에 값넣기
			if(com.equals("enter")) {
				set.add(name);
			}
			//enter가 아니면 leave이므로 해쉬에서 빼기
			else {
				set.remove(name);
			}
		}
		
		//리스트에 해쉬값 넣기
		Iterator it = set.iterator();
		while(it.hasNext()) {
			arr.add((String) it.next());
		}
		
		//배열 역순 정렬
		arr.sort(Collections.reverseOrder());
		
		//역순으로 정렬한 배열들 출력
		for(int i=0; i<arr.size();i++) {
			bw.write(arr.get(i)+"\n");
		}
		bw.flush();
		bw.close();
	}
	
}

---

주의

 

arr.sort(Collections.reverseOrder()); //역순으로 정렬하는 코드

 

string은 문자열이라서 객체로 인식하므로 ==로 비교하면 안되고 equals함수를 이용하여 비교하여야 합니다.