Golang Algorithm - 가장 가까운 같은 글자 (프로그래머스)

 

 

 

 

가장 가까운 같은 글자
문제 설명
문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s="banana"라고 할 때,  각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

b는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
n은 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 자신보다 두 칸 앞에 a가 있습니다. 이는 2로 표현합니다.
n도 자신보다 두 칸 앞에 n이 있습니다. 이는 2로 표현합니다.
a는 자신보다 두 칸, 네 칸 앞에 a가 있습니다. 이 중 가까운 것은 두 칸 앞이고, 이는 2로 표현합니다.
따라서 최종 결과물은 [-1, -1, -1, 2, 2, 2]가 됩니다.

문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제한사항
1 ≤ s의 길이 ≤ 10,000
s은 영어 소문자로만 이루어져 있습니다.

입출력 예
s result
"banana" [-1, -1, -1, 2, 2, 2]
"foobar" [-1, -1, 1, -1, -1, -1]

 

 

 

package main

import "fmt"

func closedNumber(v rune, i int, s string) int {
	count := 0
	answer := 0
	for j := i; j >= 0; j-- {
		count += 1

		if i != j && string(v) == string(s[j]) {
			answer = count - 1

			break
		} else {
			answer = -1
		}

	}
	return answer
}

func solution(s string) []int {
	arr := []int{}
	for i, v := range s {
		arr = append(arr, closedNumber(v, i, s))
	}

	return arr
}

func solution1(s string) []int {
	m := make(map[string]int)
	var res []int

	for i := range s {
		val, ok := m[string(s[i])]
		fmt.Println("val", val, "ok", ok, "!ok", !ok, "i", i, m)
		if !ok {
			m[string(s[i])] = i

			res = append(res, -1)
		} else {
			m[string(s[i])] = i
			
			res = append(res, i-val)
		}
	}
	return res
}

func main() {
	s := "banana"
	// s := "foobar"
	solution1(s)
}