코딩 테스트 및 알고리즘/leetcode for google

leetcode hard : Maximum Deletions on a String

띠리링구 2022. 10. 6. 04:37

https://leetcode.com/problems/maximum-deletions-on-a-string/

 

흥미로운 문제다

내 힘으론 못풀어서 discussion을 참고했다. (참고가 아니라 그냥 갖다 배꼈..)

class Solution:
    def deleteString(self, s: str) -> int:
        if len(set(s)) == 1 : return len(s)
        common = [[0] * (len(s) + 1) for _ in range(len(s) + 1)]
        dp = [1] * len(s)
        for i in range(len(s)-1 , -1, -1):
            for j in range(i + 1, len(s)):
                if s[i] == s[j]:
                    common[i][j] = common[i+1][j+1] + 1
                if common[i][j] >= j - i:
                    dp[i] = max(dp[i], dp[j] + 1)
        return dp[0]

첫줄 if len(set(s))는 스트링의 모든 알파벳이 같은 특수 케이스를 처리하기 위함이다. 약간 꼼수임 이거! 이거 없으면 TLE난다 ㅋㅋ;

common은 longest common substring을 구하는거다. common[i][j]는 s[i:]와 s[j:]가 몇 글자 같은지를 나타낸다.

s[i]와 s[j]가 같으면 common[i][j]에 common[i+1][j+1]+1을 대입하는 식으로 계산한다. 매우 clever한듯..

dp[i]는 s[i:]의 maximum deletions on a String을 나타낸다. common[i][j]가 j-i 이상이면 j-i만큼을 잘라낼수있는거니까 dp[j] + 1이 될수도 있고 dp[i] (안잘라내고 그냥 전체 string 다 없애기)가 될 수도 있다. 둘 중 더 큰값으로 업데이트한다.