7785 회사에 있는 사람

문제

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

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

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

입력

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

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

출력

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

예제 입력 1 복사

4
Baha enter
Askar enter
Baha leave
Artem enter

예제 출력 1 복사

Askar
Artem

 

코드

 

이름을 받아서 배열리스트에 넣고 역순으로 정렬했다.

이름이 바뀔때 이전 사람의 마지막 state를 받아와서 enter면 아직 회사에 잇는거니 결과값에 추가하고

마지막 사람이 기존 사람과 이름이 다르면 결과값에 추가되지 않았으니

결과값에 포함됬는지를 Person클래스에서 equals 메서드를 재정의해서 가져오고

enter면 따로 넣어주는 방식으로 코드를 작성해서

통과는 했지만 훨씬 나은 방법이 있었다....

import java.util.*;

class Person implements Comparable<Person>{
    String name;
    String state;

    public Person(String name, String state){
        this.state = state;
        this.name = name;
    }

    @Override
    public int compareTo(Person o) {
        return o.name.compareTo(name);
    }

    @Override
    public boolean equals(Object obj) {
        Person person = (Person) obj;
        if (person.name.equals(this.name)) return true;
        else return false;
    }
}

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        int n = in.nextInt();
        ArrayList<Person> arr = new ArrayList<>();
        ArrayList<String> res = new ArrayList<>();

        for (int i = 0; i < n; i++){
            String name = in.next();
            String state = in.next();
            arr.add(new Person(name, state));

        }

        Collections.sort(arr);

        String startName = arr.get(0).name;

        for (int i = 1; i < n; i++) {
            if (arr.get(i).name.equals(startName)) continue;
            else {
                startName = arr.get(i).name;
                if (arr.get(i-1).state.equals("enter")) res.add(arr.get(i-1).name);
            }
        }

        if (!arr.contains(startName))
            if (arr.get(n-1).state.equals("enter")) res.add(arr.get(n-1).name);

        for (String x : res) System.out.println(x);

    }
}

 

강의에서 Set 자료구조를 이용하면 시간복잡도가 log(N) 단위이니 그걸 사용하는 방식을 보았다

사람이 들어올때마다 for문으로 전체 리스트를 순회하면서 이 사람이 있는지 여부를 판별하는 방법을 생각은 햇는데

시간복잡도 N^2라 쓰지 않았지만 같은 방법도 Set을 이용하면 가능했다!

 

Set에는 Hashset, Treeset 이 있는데 Treeset은 이진탐색트리 구조라 정렬도 된 상태로 들어온다

Set을 사용해서 코드를 작성하니 훨씬 간결해졌다..

import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        TreeSet<String> res = new TreeSet<>();

        for (int i = 0; i < n; i++){
            String name = in.next();
            String state = in.next();
            if (state.equals("enter")) res.add(name);
            else res.remove(name);
        }

        String[] result = res.toArray(res.toArray(new String[res.size()]));
        Arrays.sort(result, Collections.reverseOrder());

        for (String x : result) System.out.println(x);
    }
}