[BaekJoon] 7785번 회사에 있는 사람 문제풀이 (Success)

7785번: 회사에 있는 사람

 

☕Language: Java

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        int num = Integer.parseInt(br.readLine());
 
        Map<String, String> map = new HashMap<>();
        for (int i = 0; i < num; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String name = st.nextToken();
            String status = st.nextToken();
 
            if (!map.containsKey(name)) {
                map.put(name, status);
            } else {
                map.remove(name);
            }
        }
 
        List<String> keyList = new ArrayList<>(map.keySet());
        keyList.sort((s1, s2) -> s2.compareTo(s1));
 
        for (String s : keyList) {
            bw.write(s + "\n");
        }
 
        bw.flush();
        bw.close();
    }
 
}
 

🤔 해설

1. map

    - key: name, value: status(enter or leave)

    - if 조건문

        - 회사에 동명이인은 없는 것으로 전제, 만일 동일한 이름이 입력되면 map에서 제거

 

2. keyList

    - 퇴근하지 않은 사람의 이름을 출력하기 위해 keySet으로만 구성된 List 선언

    - sort((s1, s2) -> s2.compareTo(s1)): 내림차순 정렬

 

😮 이 외의 풀이

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int num = Integer.parseInt(br.readLine());
        Map<String, String> map = new HashMap<>();
        for (int i = 0; i < num; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String name = st.nextToken();
            String status = st.nextToken();
            if (!map.containsKey(name)) {
                map.put(name, status);
            } else {
                map.remove(name);
            }
        }
        List<String> keyList = new ArrayList<>(map.keySet());
 
        Collections.sort(keyList, Collections.reverseOrder());
 
        for (String s : keyList) {
            bw.write(s + "\n");
        }
 
        bw.flush();
        bw.close();
    }
 
}
 

- 정렬: Collections.sort() + Collections.reverseOrder() 사용

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int N = Integer.parseInt(br.readLine());
 
        Set<String> set = new HashSet<>();
 
        for (int i = 0; i < N; i++) {
            String[] input = br.readLine().split(" ");
            String name = input[0];
            String status = input[1];
 
            if (set.contains(name)) {
                set.remove(name);
            } else {
                set.add(name);
            }
 
        }
 
        List<String> list = new ArrayList<>();
        Iterator<String> it = set.iterator();
 
        while (it.hasNext()) {
            list.add(it.next());
        }
 
        Collections.sort(list, Collections.reverseOrder());
 
        for (String name : list) {
            bw.write(name + "\n");
        }
 
        bw.flush();
        bw.close();
    }
 
}
 

- HashSet

   - 같은 이름이 없을 경우에는 set에 추가하고, 있을 경우에는 set에서 제거

- List

    - Collections.sort 대상은 list만 가능하므로 Iterator를 통해서 set 요소를 list에 저장

- Iterator

    - it.hasNext(): 현재 반복자가 다음 요소가 있는지 여부를 확인

    - it.next(): 현재 반복자가 가리키고 있는 요소를 반환하고, 반복자를 다음 요소로 이동

 

 

 

🔗 소스 코드
HJ0216/TIL/BOJ

 

📚 참고 자료

[백준 7785번] 회사에 있는 사람 (java)