[20291] 파일 정리 in Java

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

20291번: 파일 정리

---

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Iterator;
import java.util.Map;
import java.util.TreeMap;

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());
		Map<String,Integer> map=new TreeMap(); // 자동 오름차순 정렬을 위해 TreeMap 사용
		for (int i = 0; i < n; i++) {
			String temp=br.readLine(); // 한줄 입력받기
			temp=temp.substring(temp.indexOf(".")+1); // 확장자 떼어내기
			if(map.get(temp)==null) { // 확장자가 딕셔너리에 등록이 안 되어 있는 경우
				map.put(temp, 1); // 등록
			}else { // 확장자가 이미 딕셔너리에 등록이 되어 있는 경우
				map.put(temp, map.get(temp)+1); // +1
			}
		}
		for (Map.Entry<String, Integer> element : map.entrySet()) {
			System.out.println(element.getKey()+ " " + element.getValue());
		}
	}
}

 

(참고)

TreeMap 형태로 선언한 뒤

put을 이용하여 데이터를 넣으면,

자동으로 오름차순 정렬이 된다.