BOJ / 20291번 / 파일 정리 [Go][Python3]

👩🏻‍💻 문제

20291번: 파일 정리

---

✍🏻 풀이

🎨 Go

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

import (
	"bufio"
	"fmt"
	"os"
	"sort"
	"strings"
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	writer := bufio.NewWriter(os.Stdout)
	defer writer.Flush()

	var n int
	fmt.Fscanln(reader, &n)

	extMap := map[string]int{}
	for i := 0; i < n; i++ {
		var filepath string
		fmt.Fscanln(reader, &filepath)
		ext := strings.Split(filepath, ".")[1]
		extMap[ext]++
	}

	sortedKeys := []string{}
	for key := range extMap {
		sortedKeys = append(sortedKeys, key)
	}
	sort.Strings(sortedKeys)

	for _, key := range sortedKeys {
		fmt.Fprintf(writer, "%s %d\n", key, extMap[key])
	}
}

🎨 Python3

# https://www.acmicpc.net/problem/20291
import sys

if __name__ == "__main__":
    n = int(sys.stdin.readline())
    ext_dict = {}
    for i in range(n):
        _, ext = sys.stdin.readline().rstrip().split(".")
        if ext_dict.get(ext):
            ext_dict[ext] += 1
        else:
            ext_dict[ext] = 1
    for key in sorted(ext_dict.keys()):
        print('{} {}'.format(key, ext_dict[key]))