https://www.acmicpc.net/problem/7785
문제
> 상근이는 세계적인 소프트웨어 회사 기글에서 일한다.
> 이 회사의 가장 큰 특징은 자유로운 출퇴근 시간이다.
> 따라서, 직원들은 반드시 9시부터 6시까지 회사에 있지 않아도 된다.
> 각 직원은 자기가 원할 때 출근할 수 있고, 아무때나 퇴근할 수 있다.
> 상근이는 모든 사람의 출입카드 시스템의 로그를 가지고 있다.
> 이 로그는 어떤 사람이 회사에 들어왔는지, 나갔는지가 기록되어져 있다.
> 로그가 주어졌을 때, 현재 회사에 있는 모든 사람을 구하는 프로그램을 작성하시오.접근
맵을 사용하여 회사에 남아있는 사람을 관리한다.
입력받는 로그에 따라서 enter가 들어오면 맵에 추가하고, leave가 들어오면 map에서 제거한다.
추가로 조건에서 역순으로 출력하라고 했으므로 키값을 자동으로 정렬해주는 Treemap을 사용하는데 이는 오름차순이라 Collections.reversorder로 선언해준다.
문제해결
> n에 출입기록수를 입력받고 Treemap을 reverseOrder키값을 이름(String)으로 가지도록 선언한다.
> 출입 기록을 입력받으며 이름과 상태로 나눈뒤, 상태가 leave면 맵에서 제거, 아니면 맵에 value 1로 추가한다.
> Treemap으로 이미 정렬되어있기 때문에 m.keyset을 사용하여 map에 남은 모든 키 값을 출력한다.코드
import java.io.*;
import java.util.*;
import java.lang.*;
public class Main {
//7785번 회사에 있는 사람
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
Map<String, Integer> m = new TreeMap<>(Collections.reverseOrder());
for(int i = 0; i < n; i++) {
StringTokenizer st = new StringTokenizer(br.readLine());
String name = st.nextToken();
String cmd = st.nextToken();
if(cmd.equals("leave")) {
m.remove(name);
continue;
}
m.put(name, 1);
}
for(String key : m.keySet()) System.out.print(key + '\n');
}
}
후기
문제의 역순 출력 조건을 못봐서 자꾸 예제출력이랑 다르게 나왔다. Treemap이 우선순위 큐처럼 자동 정렬기능이 있는건 알았지만 추가로 역순으로 지정할 수 있는걸 알게 되었다. 또 keyset으로 키값만 가져올 수 있는 명령어가 있다는것도 알게 되었다.
