15-H0ngJu

[H0ngJu]

🔗 문제 링크

A와 B

✔️ 소요된 시간

2H

✨ 수도 코드

그리디 연습하려고 이 문제를 골랐습니닷 ~!

문제 요약
아래의 규칙에 따라 S를 T로 만들 수 있으면 1을, 없으면 0을 반환하라@!
규칙 1. 문자열 뒤에 A를 추가한다.
규칙 2. 문자열을 뒤집고 뒤에 B를 추가한다.

처음에는 문제 조건대로 S를 T로 만들려고 접근을 했습니다.

근데, 문제에서 제시한 테스트 케이스만 보아도,
S를 T로 만들기 위한 조건과 로직을 찾는게 진짜 만만치 않았습니다..

꾸역꾸역 구현한다는 생각으로 코드를 작성해보았지만,
장렬히 전사했습니다... 🪦🪦

사실 아래의 코드를 작성할 때는 어떤 로직이 문제인지 파악이 안돼서 그냥 레퍼런스를 참고했습니다..

항상 PR 작성할 때 쯤 되면 왜 답이 안나왔는지, 어떤 로직이 틀렸는지 눈에 보이는걸까요 .. ㅠㅜ

누더기 코드

import sys

def input(): return sys.stdin.readline().rstrip()

S = list(input())
T = list(input())

def solution(cur, target, idx):
    print(cur, target, idx)
    if len(cur) == len(target): # 문자열 길이가 같아지면 무조건 답이거나, 아니거나 둘 중 하나
        if cur == target:
            return 1
        else: return 0
    
    elif len(cur) < len(target): # 문자열 길이가 target보다 작을 때만 실행하면 됨
        if idx+1 < len(target): 
            if target[idx+1] == "B":
                cur.reverse()
                cur.append("B")
                idx += 1
                return solution(cur, target, idx)
            elif target[idx+1] == "A":
                cur.append("A")
                idx += 1
                return solution(cur, target, idx)
            else:
                return 0
        else:
            if target[idx] == "B":
                cur.reverse()
                cur.append("B")
                return solution(cur, target, idx)
            elif target[idx] == "A":
                cur.append("A")
                return solution(cur, target, idx)
            else:
                return 0

    else:
        return 0

print(solution(S, T, len(S)))

일단 찾아본 레퍼런스에서 아래의 힌트를 얻었습니다.

'S에서 T를 추적하는 것보다, T에서 S를 역추적 하는게 더 효율적이다!'

그렇게 해서 완성된 최종 수도 코드는 아래와 같습니당

1. S의 길이와 T의 길이가 같아질 때까지 아래를 수행
   1. T의 맨뒤 글자가 A이면 규칙 1에 의해 생성되었으므로, T에서 pop하고 다시 재귀
   2. T의 맨뒤 글자가 B이면 규칙 2에 의해 생성되었으므로, T에서 pop하고 reverse를 해준 뒤 다시 재귀
2. S의 길이와 T의 길이가 같아지면 S == T인지 확인
   1. S == T이면 return 1
   2. S != T이면 return 0
3. 초기값 S가 T보다 긴 경우(문제 성립 X) 케이스에 대하여 return 0

.
.
.

그런데 계속해서 런타임에러(RecursionError) 가 발생하더라고요 ...

혹시 계속해서 재귀되는 케이스가 있나 싶어서 질문게시판 예시들도 다 돌려봤으나 .. 해결을 못했습니다

그러다가

동방에 있는 @9kyo-hwang 도움으로 ..

sys.setrecursionlimit(10**6)

recursion limit을 늘렸더니 .. 통과가 되었습니다 ... 🥹🥹

후 예전에 못 풀었던 그리디 문제들도 도장깨기 갑니닷 👊👊👊👊👊

📚 새롭게 알게된 내용

파이썬 recursion error는 sys.setrecursionlimit(106)로 해결하기 ..
이거 해결하는데 한 한시간은 쳐다 본 것 같네요 🫠

[tgyuuAn]

와 저도 이 문제 S -> T로 풀려다가 gg 치고 레퍼런스 보고 T -> S로 풀었습니다..

코테에 나왔으면 틀렸을 것 같네요.....................

와퐁주 짱..........

import sys

def input(): return sys.stdin.readline().rstrip()

S = list(input())
T = list(input())

while len(T) > len(S):
    if T[-1] == "A": T.pop()
    else:
        T.pop()
        T = T[::-1]

if S == T: print(1)
else: print(0)

[tgyuuAn]

근데 이 부분 재귀로 하신 이유가 있을까요?

똑같은 구현이면 파이썬은 재귀타는게 무거워서 while문이 훨씬 빨라요!!!

저렇게 recursionlimit 설정해줄 필요도 없꾸요 하하

쨋든 쨋든.... 좋은 문제 너무 감사합니다요잇

[H0ngJu]

사실 별다른 이유는 없습니다.. 허허
처음에 함수 설계할 때 '재귀 써야지~' 해서 작성하다보니 저렇게 된 겁니다요... ㅋ.ㅋ.ㅋ.

태규님 말대로 사실 저 재귀로직을

while len(cur)<len(target)으로 해버리면 recursionlimit 설정 안해도 되어요 🥹🥹

좋은 피드백 감사하므니당 👍

[tgyuuAn]

@H0ngJu 홍쥬르 이거 혹시 https://discord.com/channels/1203381153764089958/1211331998010834945 여기 그리디로 분류해주셔서 올려주실 수 있나요 ?

개인적으로 전 머리 탁 치는 문제였는데

[SeongHoonC]

저두 재귀로 안풀고 반복문으로 풀었어요!
뭔가 비밀이 숨겨져 있을 것 같았는데 반대로 푸는게 답이었네요. 그리디도 DP 처럼 아이디어가 중요한 것 같아요..!

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val s = br.readLine()!!
    val t = br.readLine()!!

    println(if (sisT(s, t)) 1 else 0)
}

private fun sisT(s: String, t: String): Boolean {
    var now = t
    while (s.length < now.length) {
        now = if (now.last() == 'B') {
            now.substring(0, now.lastIndex).reversed()
        } else {
            now.substring(0, now.lastIndex)
        }
    }
    return s == now
}

[alstjr7437]

아 DFS로 풀때 한번씩 재귀 때문에 시간초과가 뜨는데 저가 잘못 푼거인지 할때가 있더라구요...

저도 다른 분들과 같은 방식으로 재귀를 사용하지 않고 푸는 방법을 생각했었는데 재귀로도 한번 풀어봐야겠네요!!