백준 1302번 - 베스트셀러 / Go

문제: https://www.acmicpc.net/problem/1302

1302번: 베스트셀러

문제 설명

문제

김형택은 탑문고의 직원이다. 김형택은 계산대에서 계산을 하는 직원이다. 김형택은 그날 근무가 끝난 후에, 오늘 판매한 책의 제목을 보면서 가장 많이 팔린 책의 제목을 칠판에 써놓는 일도 같이 하고 있다.

오늘 하루 동안 팔린 책의 제목이 입력으로 들어왔을 때, 가장 많이 팔린 책의 제목을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 오늘 하루 동안 팔린 책의 개수 N이 주어진다. 이 값은 1,000보다 작거나 같은 자연수이다. 둘째부터 N개의 줄에 책의 제목이 입력으로 들어온다. 책의 제목의 길이는 50보다 작거나 같고, 알파벳 소문자로만 이루어져 있다.

출력

첫째 줄에 가장 많이 팔린 책의 제목을 출력한다. 만약 가장 많이 팔린 책이 여러 개일 경우에는 사전 순으로 가장 앞서는 제목을 출력한다.

예제 입력 1 복사

5
top
top
top
top
kimtop

예제 출력 1 복사

top

예제 입력 2 복사

9
table
chair
table
table
lamp
door
lamp
table
chair

예제 출력 2 복사

table

예제 입력 3 복사

6
a
a
a
b
b
b

예제 출력 3 복사

a

예제 입력 4 복사

8
icecream
peanuts
peanuts
chocolate
candy
chocolate
icecream
apple

예제 출력 4 복사

chocolate

예제 입력 5 복사

1
soul

예제 출력 5 복사

soul
​

출처

• 문제를 번역한 사람: baekjoon
• 데이터를 추가한 사람: sukwoo0711

알고리즘 분류

• 자료 구조
• 문자열
• 정렬
• 해시를 사용한 집합과 맵

 

---

정답

package main

import (
	"bufio"
	"os"
	"sort"
	"strconv"
	"strings"
	"testing"
)

func main() {
	rd := bufio.NewReader(os.Stdin)
	wr := bufio.NewWriter(os.Stdout)

	n, _ := strconv.Atoi(scan1302(rd))

	books := make([]string, 0, n)
	for i := 0; i < n; i++ {
		books = append(books, scan1302(rd))
	}

	_, _ = wr.WriteString(solution1302(books))
	_ = wr.Flush()
}

func solution1302(books []string) string {
	count := make(map[string]int)
	for _, name := range books {
		count[name]++
	}

	maxCount := -1
	maxBooks := make([]string, 0, len(count))
	for name, num := range count {
		if num > maxCount {
			maxBooks = append(maxBooks[:0], name)
			maxCount = num
		} else if num == maxCount {
			maxBooks = append(maxBooks, name)
		}
	}

	sort.Slice(maxBooks, func(i, j int) bool {
		return maxBooks[i] < maxBooks[j]
	})
	return maxBooks[0]
}

func scan1302(rd *bufio.Reader) string {
	str, _ := rd.ReadString('\n') // 여기서 text는 마지막에 줄바꿈 문자를 포함하므로
	str = strings.TrimSpace(str)  // 줄바꿈 문자를 제거해야 함
	return str
}

func Benchmark1302(b *testing.B) {
	for i := 0; i < b.N; i++ {
		solution1302([]string{"icecream", "peanuts", "peanuts", "chocolate", "candy", "chocolate", "icecream", "apple"})
	}
}
func Test_solution1302(t *testing.T) {
	type args struct {
		books []string
	}
	tests := []struct {
		name       string
		args       args
		wantResult string
	}{
		{name: "", args: args{books: []string{"top", "top", "top", "top", "kimtop"}}, wantResult: "top"},
		{name: "", args: args{books: []string{"table", "chair", "table", "table", "lamp", "door", "lamp", "table", "chair"}}, wantResult: "table"},
		{name: "", args: args{books: []string{"icecream", "peanuts", "peanuts", "chocolate", "candy", "chocolate", "icecream", "apple"}}, wantResult: "chocolate"},
		{name: "", args: args{books: []string{"soul"}}, wantResult: "soul"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if gotResult := solution1302(tt.args.books); gotResult != tt.wantResult {
				t.Errorf("solution1302() = %v, want %v", gotResult, tt.wantResult)
			}
		})
	}
}

 

---

풀이

 map을 이용해서 책 별 판매량을 확인한 다음, 최고 판매량 중 사전 순으로 앞서는 제목을 출력하면 되는 문제이다.

 

 Golang에서는 map에 해당 키가 존재하지 않아도 "++"등의 로직을 처리할 수 있기 때문에 처음에 키가 존재하는지 확인을 할 필요가 없어서 간단히 해결 가능하다.

 

 map에서 최댓값을 뽑아내는 기능이 있다면 편리할 것 같다.