개발자가 되

프로그래머스 Lv.1

Lv.1 가장 가까운 같은 글자

dltjdud 2025. 2. 7. 15:41

문제설명

문자열 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]

 

입출력 예 설명

입출력 예 #1
지문과 같습니다.

입출력 예 #2
설명 생략

 

풀이

 

JAVA

import java.util.HashMap;
import java.util.Map;

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        Map<Character, Integer> map = new HashMap<>();
        for (int i = 0; i < s.length(); i++) {
            if (!map.containsKey(s.charAt(i))) {
                answer[i] = -1;
                map.put(s.charAt(i), i);
                continue;
            }

            answer[i] = i - map.get(s.charAt(i));
            map.put(s.charAt(i), i);
        }

        return answer;
    }
}

*다른 코드

import java.util.*;

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        HashMap<Character,Integer> map = new HashMap<>();
        for(int i=0; i<s.length();i++){
            char ch = s.charAt(i);
            answer[i] = i-map.getOrDefault(ch,i+1);
            map.put(ch,i);
        }
        return answer;
    }
}

 

Python

def solution(s):
    answer = []
    dic = dict()
    for i in range(len(s)):
        if s[i] not in dic:
            answer.append(-1)
        else:
            answer.append(i - dic[s[i]])
        dic[s[i]] = i

    return answer

*다른 코드

def solution(s):
    answer = []
    dic = {}

    for i in range(len(s)):
        if s[i] not in dic:
            dic[s[i]]= i
            answer.append(-1)
        else:
            answer.append(i-dic[s[i]])
            dic[s[i]] = i
    return answer

 

JS

function solution(s) {
    let obj={}
    let answer=[]
    for(let i =0;i<s.length;i++){
        if(obj[s[i]]===undefined){
            answer.push(-1)
            obj[s[i]]=i
        }else{
            answer.push(i-obj[s[i]])
            obj[s[i]]=i
        }
    }
    return answer;
}

*다른 코드

function solution(s) {
    const hash={};

    return [...s].map((v,i)=>{
        let result = hash[v] !== undefined ? i - hash[v] : -1;
        hash[v] = i;
        return result;
    });
}

 

'프로그래머스 Lv.1' 카테고리의 다른 글

Lv.1 기사단원의 무기  (0) 2025.02.08
Lv.1 명예의 전당 (1)  (0) 2025.02.08
Lv.1 크기가 작은 부분문자열  (0) 2025.02.07
Lv.1 카드 뭉치  (0) 2025.02.07
Lv.1 덧칠하기  (0) 2025.02.07