본문 바로가기
IT/알고리즘(Algorithm)

[JAVA]백준 11650번 문제 풀이

by 공부하는개미 2022. 11. 24.

# 문제 출제 사이트

https://www.acmicpc.net/problem/11650

 

11650번: 좌표 정렬하기

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

www.acmicpc.net

 

# 문제

2차원 평면 위의 점 N개가 주어진다.

좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

 

# 입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다.

둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다.

(-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

 

# 출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

 

 

# 제출한 소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    int n = Integer.parseInt(br.readLine());

    String[] strArr = new String[n];
    for ( int i = 0; i < n; i++ ) {
        strArr[i] = br.readLine();
    }

    Arrays.sort(strArr, new Comparator<String>() {
        @Override
        public int compare(String str1, String str2) {
            String[] str1Arr = str1.split(" ");
            String[] str2Arr = str2.split(" ");
            
            if ( !str1Arr[0].equals(str2Arr[0])) {
                return Integer.parseInt(str1Arr[0]) - Integer.parseInt(str2Arr[0]);
            } else {
                return Integer.parseInt(str1Arr[1]) - Integer.parseInt(str2Arr[1]);
            }
        }
    });
    
    
    for (String str : strArr ) {
        System.out.println(str);
    }
  }
}
  • 다른 자료 참고 안하고 직접 풀어본 방법
  • Arrays.sort에서 compare를 @override해서 사용해봤다.
  • 시간 2404ms 로 상당히 느리게 나왔다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    int n = Integer.parseInt(br.readLine());

    int[][] intArr = new int[n][2];
    for ( int i = 0; i < n; i++ ) {
        String[] tempArr = br.readLine().split(" ");
        intArr[i][0] = Integer.parseInt(tempArr[0]);
        intArr[i][1] = Integer.parseInt(tempArr[1]);
    }

    Arrays.sort(intArr, (e1, e2) -> {
        if ( e1[0] != e2[0] ) {
            return e1[0] - e2[0];
        } else {
            return e1[1] - e2[1];
        }
    });

    for ( int j = 0; j < n; j++ ) {
        System.out.println(intArr[j][0] + " " + intArr[j][1]);
    }
  }
}
  • 2차원 int 배열로 번경해서 해봤다.
  • Arrays.sort 부분에 람다식을 사용해서 문제를 풀어봤다.
  • 시간 1580ms 로 전에 제출한 것보다 훨씬 빨라졌다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer st;
    int n = Integer.parseInt(br.readLine());

    int[][] intArr = new int[n][2];
    for ( int i = 0; i < n; i++ ) {
        st = new StringTokenizer(br.readLine());
        intArr[i][0] = Integer.parseInt(st.nextToken());
        intArr[i][1] = Integer.parseInt(st.nextToken());
    }

    Arrays.sort(intArr, (e1, e2) -> {
        if ( e1[0] != e2[0] ) {
            return e1[0] - e2[0];
        } else {
            return e1[1] - e2[1];
        }
    });

    for ( int j = 0; j < n; j++ ) {
        System.out.println(intArr[j][0] + " " + intArr[j][1]);
    }
  }
}

 

  • 한줄에 입력되는 문자열을 split(" ") 함수를 사용해서 공백을 잘랐는데 StringTokenizer를 사용해 봤다.
  • 시간은 1400ms 가 나왔다.

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
  public static void main(String[] args) throws IOException {
      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
      StringTokenizer st;
      int n = Integer.parseInt(br.readLine());

      int[][] intArr = new int[n][2];
      for ( int i = 0; i < n; i++ ) {
          st = new StringTokenizer(br.readLine());
          intArr[i][0] = Integer.parseInt(st.nextToken());
          intArr[i][1] = Integer.parseInt(st.nextToken());
      }

      Arrays.sort(intArr, (e1, e2) -> {
          if ( e1[0] != e2[0] ) {
              return e1[0] - e2[0];
          } else {
              return e1[1] - e2[1];
          }
      });

      StringBuilder sb = new StringBuilder();
      for(int j = 0 ; j< n ; j++) {
          sb.append(intArr[j][0] + " " + intArr[j][1]).append('\n');
      }
      System.out.println(sb);
  }
}
  • 프린트 함수를 사용하지 않고 StringBuilder를 사용해서 출력을 해봤다.
  • 시간 744ms 가 나왔다.

 

# 참고자료

  • 블로그 Stranger's LAB => 링크

 

반응형