99클럽 코테 스터디 36일차 TIL + 스택

문제

2390. Removing Stars From a String 

https://leetcode.com/problems/removing-stars-from-a-string/

 

코드

class Solution(object):
    def removeStars(self, s):
        s = list(s)

        while '*' in s:
            for i in range(len(s)-1):
                if s[i + 1] == '*':
                    s.pop(i+1)
                    s.pop(i)
                    break

        return ''.join(s)

 

풀이

s = list(s)

매개변수로 받은 문자열 s를 리스트화 시킨다.

ex ) "leet**cod*e" ➡️ ['l', 'e', 'e', 't', '*', '*', 'c', 'o', 'd', '*', 'e']

 

while '*' in s:
    for i in range(len(s)-1):
        if s[i + 1] == '*':
            s.pop(i+1)
            s.pop(i)
            break

s배열에 '*'이 없어질때까지 다음의 작업을 계속 반복한다.

1. s의 인덱스 i번째 원소의 다음 원소가 '*'이라면
2. 인덱스 i번째 원소의 다음 원소인 '*'을 pop하고
3. 지금 방문중인 원소 인덱스 i번째 원소도 pop한다.
4. 한번의 삭제 작업을 했다면 break하고 다시 1번작업을 반복한다.

 

return ''.join(s)

s의 원소들을 하나의 문자열로 합해서 반환한다.

 

다른코드

class Solution(object):
    def removeStars(self, s):
        stack = []
       
        for c in s:
            if c == '*' and stack:
                stack.pop()
            else:
                stack.append(c)

        return ''.join(stack)

 

풀이

이전 코드보다 스택의 개념이 잘 적용된 코드이다.

 

stack = []

빈 stack 배열을 생성한다.

 

for c in s:
    if c == '*' and stack:
        stack.pop()
    else:
        stack.append(c)

입력 받았던 s 문자열을 한글자씩(c) 방문한다.

만약 c가 '*'이고 스택에 원소가 있다면, 스택에 있는 원소중 제일 마지막 원소를 pop한다.

c가 '*'이 아니거나 스택에 원소가 있다면, 스택에 c를 차곡차곡 추가한다.

 

return ''.join(stack)

스택에 있는 원소들을 하나의 문자열로 합해서 반환한다.

 

새롭게 배운점 / 느낀점

• 간단한 문제여도 키워드를 적용하여 문제를 풀 줄 아느냐 모르냐 차이는 큰 것 같다.
• 쉬운 구현 문제여도 여러가지 키워들를 대입해서 문제를 풀어보며 다양하게 접근하는 관점을 길러야겠다.

 

다른 의견이나 궁금하신게 있다면 편하게 댓글 달아주세요 :)