백준 1431 시리얼 번호(Java)

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

1431번: 시리얼 번호

---

 

해석 및 팁

 

---

 

이 문제는 Comparator을 이용해서 조건에 맞춰서 정렬하면 됩니다.

 

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();
    
    int n = sc.nextInt();
    String[] arr = new String[n];

    for(int i = 0; i < n; i++) arr[i] = sc.next();

    Arrays.sort(arr, new Comparator<String>() {
      public int compare(String s1, String s2) {
        if(s1.length() == s2.length()) {
          int sum1 = 0;
          int sum2 = 0;
          for(int i = 0; i < s1.length(); i++) {
            if(s1.charAt(i) >= '0' && s1.charAt(i) <= '9') sum1 += s1.charAt(i) - '0';
          }
          for(int i = 0; i < s2.length(); i++) {
            if(s2.charAt(i) >= '0' && s2.charAt(i) <= '9') sum2 += s2.charAt(i) - '0';
          }
          if(sum1 == sum2) return s1.compareTo(s2);
          else return sum1 - sum2;
        }
        else return s1.length() - s2.length();
      }
    });

    for(int i = 0; i < n; i++) sb.append(arr[i]+"\n");

    System.out.println(sb);
  }
}