[Programmers/Go] 가운데 글자 가져오기

1. 문제

단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다.

제한사항

• s는 길이가 1 이상, 100이하인 스트링입니다.

2. 풀이

// 가운데 글자 가져오기
package centerletter

func Solution(s string) string {
	var answer string
	length := len(s)

	if length%2 == 0 {
		answer = s[length/2-1 : length/2+1]
	} else {
		answer = s[length/2 : length/2+1]
	}

	return answer
}

3. 테스트

package centerletter

import "testing"

func TestCenterletter(t *testing.T) {
	input := []string{"abcde", "qwer"}
	expect := []string{"c", "we"}

	for i := range input {
		result := Solution(input[i])
		if result != expect[i] {
			t.Errorf("Test%d: Wrong result", i+1)
			t.Error("expect:", expect[i])
			t.Error("result:", result)
		}
	}
}

programmers.co.kr/learn/courses/30/lessons/12903

코딩테스트 연습 - 가운데 글자 가져오기