[BOJ] 정렬 단계3 - JAVA

두 문제를 끝으로 단계별 정렬 문제는 다 풀었다.

정렬 단계

• 단어 정렬
• 나이순 정렬

 ▷ 단어 정렬 

www.acmicpc.net/problem/1181

1181번: 단어 정렬

단어를 길이순서로 정렬하고 길이가 같은 때는 사전 순으로 정렬해야 하는데 다른 사람것을 보니까 람다로 푼 사람도 있었다.

사실 hashmap 넣어서 푸는 것보다는 메모리를 아낄 수 있는 방법이여서 그게 좋은데 아직까지 람다를 제대로 공부해본적이 없어서 모르겠다. 람다를 공부하게 되면 그때 다시 한번 풀어봐야지...

▼ 아래는 맨 처음 제출한 코드이다.

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

public class Main{
    public static void main(String[] args) throws IOException{
        Map<Integer,ArrayList<String>> hp = new HashMap<Integer,ArrayList<String>>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int testCase = Integer.parseInt(br.readLine().trim());
        
        for(int i = 0; i<testCase; i++){
            ArrayList<String> arr ;
            String word = br.readLine().trim();
            int leng = word.length();
            if(hp.containsKey(leng)){
                arr = hp.get(leng);
            }else{
                arr = new ArrayList<>();
            }
            arr.add(word);
            Collections.sort(arr);
            hp.put(leng, arr);
        }
        
        Object[] mapKeys = hp.keySet().toArray();
        Arrays.sort(mapKeys);
        
        for(Object key : mapKeys){
            for(String word : hp.get((int)key)){
                bw.write(word+"\n");
            }
        }
        bw.flush();
        bw.close();
        br.close();
    }
}

틀린 이유는 입력값 중 같은 단어는 한번만 입력되어야 하는데 이부분을 빼먹었기 때문이다.

▼ if문 하나 추가 하니 정답이였다.

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

public class Main{
    public static void main(String[] args) throws IOException{
        Map<Integer,ArrayList<String>> hp = new HashMap<Integer,ArrayList<String>>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int testCase = Integer.parseInt(br.readLine().trim());
        
        for(int i = 0; i<testCase; i++){
            ArrayList<String> arr ;
            String word = br.readLine().trim();
            int leng = word.length();
            if(hp.containsKey(leng)){
                arr = hp.get(leng);
            }else{
                arr = new ArrayList<>();
            }
            if(!arr.contains(word))
                arr.add(word);
            Collections.sort(arr);
            hp.put(leng, arr);
        }
        
        Object[] mapKeys = hp.keySet().toArray();
        Arrays.sort(mapKeys);
        
        for(Object key : mapKeys){
            for(String word : hp.get((int)key)){
                bw.write(word+"\n");
            }
        }
        bw.flush();
        bw.close();
        br.close();
    }
}

 

 ▷ 나이순 정렬 

www.acmicpc.net/problem/10814

10814번: 나이순 정렬

나이순 정렬은 우선 나이순으로 정렬을 하고 나이가 같을 때는 가입순서로 배열을 해야한다. 나는 이부분을 제대로 읽지 않아서 당연히 이름 순서대로겠거니 했다. 그래서 결과적으로는 틀린 코드를 작성해버렸다.. 매번 이런식이지,,

▼ 아래는 틀린 코드이다.

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

public class Main{
    public static void main(String[] args) throws IOException{
        Map<Integer,ArrayList<String>> hp = new HashMap<Integer,ArrayList<String>>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int testCase = Integer.parseInt(br.readLine().trim());
        
        for(int i = 0; i<testCase; i++){
            ArrayList<String> arr ;
            StringTokenizer st = new StringTokenizer(br.readLine()," ");
            int age =Integer.parseInt(st.nextToken());
            String name = st.nextToken();
            if(hp.containsKey(age)){
                arr = hp.get(age);
            }else{
                arr = new ArrayList<>();
            }
            arr.add(name);
            Collections.sort(arr);
            hp.put(age, arr);
        }
        
        Object[] mapKeys = hp.keySet().toArray();
        Arrays.sort(mapKeys);
        
        for(Object key : mapKeys){
            for(String name : hp.get((int)key)){
                bw.write(key+" "+name+"\n");
            }
        }
        bw.flush();
        bw.close();
        br.close();
    }
}

나이가 같을 때는 가입된 순서로 입력하면되는것이기 때문에 이름을 담는 ArrayList를 정렬하지 않아도 된다.

▼ 아래는 맞는 코드이다.

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

public class Main{
    public static void main(String[] args) throws IOException{
        Map<Integer,ArrayList<String>> hp = new HashMap<Integer,ArrayList<String>>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int testCase = Integer.parseInt(br.readLine().trim());
        
        for(int i = 0; i<testCase; i++){
            ArrayList<String> arr ;
            StringTokenizer st = new StringTokenizer(br.readLine()," ");
            int age =Integer.parseInt(st.nextToken());
            String name = st.nextToken();
            if(hp.containsKey(age)){
                arr = hp.get(age);
            }else{
                arr = new ArrayList<>();
            }
            arr.add(name);
            hp.put(age, arr);
        }
        
        Object[] mapKeys = hp.keySet().toArray();
        Arrays.sort(mapKeys);
        
        for(Object key : mapKeys){
            for(String name : hp.get((int)key)){
                bw.write(key+" "+name+"\n");
            }
        }
        bw.flush();
        bw.close();
        br.close();
    }
}

 

develop-recode.tistory.com/21

[BOJ] 정렬 단계2 - JAVA