[Programmers/Go] 수박수박수박수박수박수?

1. 문제

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

제한 조건

• n은 길이 10,000이하인 자연수입니다.

2. 풀이

// 수박수박수박수박수박수?
package watermelon

import "strings"

func Solution(n int) string {
	str := strings.Repeat("수박", n)
	return strings.Join(strings.Split(str, "")[:n], "")
}

3. 테스트

package watermelon

import "testing"

func TestWatermelon(t *testing.T) {
	input := []int{3, 4}
	expect := []string{"수박수", "수박수박"}

	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/12922

코딩테스트 연습 - 수박수박수박수박수박수?