백준 2480번 - 주사위 세개 / Go

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

2480번: 주사위 세개

문제 설명

 

문제

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다. 

1. 같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다. 
2. 같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다. 
3. 모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.  

예를 들어, 3개의 눈 3, 3, 6이 주어지면 상금은 1,000+3×100으로 계산되어 1,300원을 받게 된다. 또 3개의 눈이 2, 2, 2로 주어지면 10,000+2×1,000 으로 계산되어 12,000원을 받게 된다. 3개의 눈이 6, 2, 5로 주어지면 그중 가장 큰 값이 6이므로 6×100으로 계산되어 600원을 상금으로 받게 된다.

3개 주사위의 나온 눈이 주어질 때, 상금을 계산하는 프로그램을 작성 하시오.

입력

첫째 줄에 3개의 눈이 빈칸을 사이에 두고 각각 주어진다. 

출력

첫째 줄에 게임의 상금을 출력 한다.

예제 입력 1 복사

3 3 6

예제 출력 1 복사

1300

예제 입력 2 복사

2 2 2

예제 출력 2 복사

12000

예제 입력 3 복사

6 2 5

예제 출력 3 복사

600

---

정답

package main

import (
	"fmt"
	"sort"
	"testing"
)

func main() {
	var a, b, c int
	_, _ = fmt.Scanln(&a, &b, &c)
	fmt.Println(solution2480(a, b, c))
}

func solution2480(a, b, c int) (result int) {
	arr := []int{a, b, c}
	sort.Ints(arr)
	a, b, c = arr[0], arr[1], arr[2]
	if a == b && b == c {
		result = 10000 + a*1000
	} else if a == b || b == c {
		result = 1000 + b*100
	} else {
		result = c * 100
	}
	return
}

func Test_solution2480(t *testing.T) {
	type args struct {
		a int
		b int
		c int
	}
	tests := []struct {
		name       string
		args       args
		wantResult int
	}{
		{
			name: "1",
			args: args{
				a: 3,
				b: 3,
				c: 6,
			},
			wantResult: 1300,
		},
		{
			name: "2",
			args: args{
				a: 2,
				b: 2,
				c: 2,
			},
			wantResult: 12000,
		},
		{
			name: "3",
			args: args{
				a: 6,
				b: 2,
				c: 5,
			},
			wantResult: 600,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if gotResult := solution2480(tt.args.a, tt.args.b, tt.args.c); gotResult != tt.wantResult {
				t.Errorf("solution() = %v, want %v", gotResult, tt.wantResult)
			}
		})
	}
}

 

---

풀이

 간단한 if문을 이용한 문제입니다. 보통 여기서 두 번째 조건을 작성할 때 코드가 길어질 텐데, 정렬을 이용하여 푼다면 가독성 좋게 짧은 코딩으로 문제를 해결할 수 있습니다.