[백준] 2170번 - 선 긋기 (java)

Baekjoon 2170 - 선 긋기 (클릭 시 이동)

문제

매우 큰 도화지에 자를 대고 선을 그으려고 한다. 선을 그을 때에는 자의 한 점에서 다른 한 점까지 긋게 된다. 선을 그을 때에는 이미 선이 있는 위치에 겹쳐서 그릴 수도 있는데, 여러 번 그은 곳과 한 번 그은 곳의 차이를 구별할 수 없다고 하자.

이와 같은 식으로 선을 그었을 때, 그려진 선(들)의 총 길이를 구하는 프로그램을 작성하시오. 선이 여러 번 그려진 곳은 한 번씩만 계산한다.

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
1 초	192 MB	8648	2982	2225	36.746%

입력

첫째 줄에 선을 그은 횟수 N(1 ≤ N ≤ 1,000,000)이 주어진다. 다음 N개의 줄에는 선을 그을 때 선택한 두 점의 위치 x, y(-1,000,000,000 ≤ x < y ≤ 1,000,000,000)가 주어진다.

출력

첫째 줄에 그은 선의 총 길이를 출력한다.

풀이

1. 두 선이 주어지면 x의 오름차순으로, 동일한 x값이 있으면 y의 오름차순으로 정렬한다. ( 우선순위 큐 이용 )
2. 서로다른 두 선 A(x1,y1) , B(x2,y2)가 있으면 y1과 x1을 비교하고 일직선으로 이루어 져 있는지 확인한다.
3. 일직선이면 그 선분의 끝점을 갱신한다.
4. 일직선이 아니라면 여지껏 저장해둔 선분의 길이를 업데이트 하고 갱신한다.

최종 소스코드

package BJ_Practice.Gold;

import java.io.*;
import java.util.*;

public class BJ_G5_2170_선_긋기 {

    static int N;
    static StringTokenizer st;

    static class Line implements Comparable<Line> {
        int start, end;

        public Line(int start, int end) {
            super();
            this.start = start;
            this.end = end;
        }

        @Override
        public int compareTo(Line o) {
            if (this.start == o.start)
                return this.end - o.end;
            return this.start - o.start;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        N = Integer.parseInt(br.readLine());
        PriorityQueue<Line> pq = new PriorityQueue<>();
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            pq.offer(new Line(start, end));
        }
        int ans = 0;
        int start = Integer.MIN_VALUE, end = Integer.MIN_VALUE;
        while (!pq.isEmpty()) {
            Line temp = pq.poll();
            if (end >= temp.start) {
                end = Math.max(temp.end, end);

            } else {
                ans += Math.abs(end - start);
                start = temp.start;
                end = temp.end;
            }
            if (pq.isEmpty()) {
                ans += Math.abs(end - start);

            }
        }
        System.out.println(ans);
    }
}