[백준] 20920 영단어 암기는 괴로워 - 문자열(String) + 정렬(Sort) + 해시맵(HashMap) JAVA

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

20920번: 영단어 암기는 괴로워

코드설명

구현 + 문자열 + 정렬 + hashmap 을 사용하여 해결합니다.

 

1. HashMap을 순회하는 코드입니다.

for(Map.Entry<String, Integer> entry : hashmap.entrySet()){
    nodeList.add(new Node(entry.getKey(), entry.getValue()));
}

 

2. 문자열의 사전순을 비교할때는 compareTo() 를 통해 진행합니다.

this.word.compareTo(other,word) > 0 이라면

this.word의 사전순이 더 크므로 this.word - other.word 사전순 값이 > 0 입니다.

그러므로 사전순값으로 더 빠르게해야하므로 return 1을 통해서 순서를 바꿔줍니다.

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.util.Map.Entry;
import java.util.stream.Collectors;


public class Main {
	
	public static int N, M;
	public static String[] arr;
	public static HashMap<String, Integer> hashmap;
	public static ArrayList<Node> nodeList = new ArrayList<>();
    public static void main(String[] args) throws IOException{
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    	
    	StringTokenizer st = new StringTokenizer(br.readLine());
    	N = Integer.parseInt(st.nextToken());
    	M = Integer.parseInt(st.nextToken());
    	
    	hashmap = new HashMap<String, Integer>();
    	arr = new String[N];
    	for(int i=0;i<N;i++) {
    		st = new StringTokenizer(br.readLine());
    		arr[i] = st.nextToken();
    		if(arr[i].length() < M) continue;
    		hashmap.put(arr[i], hashmap.getOrDefault(arr[i], 0) + 1);
    	}
    	
    	for(Map.Entry<String, Integer> entry : hashmap.entrySet()) {
//    		System.out.println(entry.getValue()+" "+entry.getKey());
    		nodeList.add(new Node(entry.getKey(), entry.getValue()));
    	}
    	
    	Collections.sort(nodeList);
    	
    	StringBuilder sb = new StringBuilder();
    	for(int i=0;i<nodeList.size();i++) {
//    		System.out.println(nodeList.get(i).word);
    		sb.append(nodeList.get(i).word).append("\n");
    	}
    	System.out.println(sb);
    	
    	
    }
    
}

class Node implements Comparable<Node>{
	String word;
	int cnt;
	public Node(String word, int cnt) {
		this.word = word;
		this.cnt = cnt;
	}
	
	@Override
	public int compareTo(Node other) {
		if(this.cnt < other.cnt) {
			return 1;
		}else if(this.cnt == other.cnt) {
			if(this.word.length() < other.word.length()) {
				return 1;
			}else if(this.word.length() == other.word.length()) {
				if(this.word.compareTo(other.word) > 0) {
					return 1;
				}else if(this.word.compareTo(other.word) == 0) {
					return 0;
				}else {
					return -1;
				}
			}else if(this.word.length() > other.word.length()) {
				return -1;
			}
		}
		else {
			return -1;
		}
		
		return -1;
	}
}

 

Comparator로 익명함수를 사용할 경우입니다. 

가능하면 Node Class 대신 배열을 사용하여 진행하고 싶었지만, String label과 int cnt를 함께 쓰기 위해서는 객체를 사용하면 좋습니다.

package Main;
 
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.StringTokenizer;
 
public class Main {
	static int N, M, S, P, A, B, X, L, R, C, n, k, m, l, K, D, T;
	static int answer = 0;
	static ArrayList<Node> node = new ArrayList<Node>();
	static HashMap<String, Integer> hm = new HashMap<>();
	static StringBuilder sb = new StringBuilder();
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		N = Integer.parseInt(st.nextToken());
		M = Integer.parseInt(st.nextToken());
		
		
		for(int i=0; i<N; i++) {
			String word = br.readLine();
			if(word.length() >= M) {
				if(hm.containsKey(word) == true) {
					int index = hm.get(word);
					node.get(index).cnt += 1;
				}else {
					hm.put(word, node.size());
					node.add(new Node(word, 1));
				}
			}
		}
		
		Collections.sort(node, new Comparator<Node>() {
			@Override
			public int compare(Node now, Node other) {
				// TODO Auto-generated method stub
				if(now.cnt != other.cnt) return -Integer.compare(now.cnt, other.cnt);
				if(now.label.length() != other.label.length()) return -Integer.compare(now.label.length(), other.label.length());
				return now.label.compareTo(other.label);
			}
		});
		
		for(Node v : node) {
//			System.out.println(v.label);
			sb.append(v.label).append('\n');
		}
		System.out.println(sb.toString());
	}
	
	static class Node {
		String label;
		int cnt;
		public Node(String label, int cnt) {
			this.label = label;
			this.cnt = cnt;
		}
	}
}