[프로그래머스] 가장 가까운 같은 글자
문제
문자열 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은 영어 소문자로만 이루어져 있습니다.
입출력 예
sresult
| "banana" | -1, -1, -1, 2, 2, 2 |
| "foobar" | -1, -1, 1, -1, -1, -1 |
풀이
문제를 정리 해보면 자신과 동일한 문자가 앞에 존재한다면 그 인덱스와 자신의 인덱스의 차이를 구하고,
그렇지 못하면 -1을 return시킨다.
따라서 map을 사용하여 이 문제를 풀었다.
map으로 해당 문자를 찾아서 존재하지 않다면, 리스트에 -1을 저장하고, 그 문자와 인덱스를 key, value로 저장한다.
문자가 존재 한다면 인덱스를 차를 리스트에 저장한다.
import java.util.*;
import java.lang.*;
class Solution {
public int[] solution(String s) {
List<Integer> list = new ArrayList<>();
Map<String, Integer> map = new HashMap<>();
for(int i = 0; i < s.length(); i++) {
String subStr = s.substring(i, i + 1);
if (map.get(subStr) == null){
list.add(-1);
} else {
list.add(i - map.get(subStr));
}
map.put(subStr, i);
}
return list.stream().mapToInt(i -> i).toArray();
}
}