Problem Solving/LeetCode

Today's Challenge

fabichoi 2022. 9. 20. 23:45

https://leetcode.com/problems/maximum-length-of-repeated-subarray/

 

Maximum Length of Repeated Subarray - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com

문제를 딱 보는 순간 n이 1000인데 O(n^3)이라서 TLE가 날 거 같아서 무식하게 풀기는 어려워 보임.
아마 슬라이딩 윈도우나 투 포인터를 써야 되는 문제인거 같은데.. 어떻게 코드 작성해야 하는지 기억안남.
그래서 결국 솔루션 엔딩 =_=

------------------------------------------

아니었다. 솔루션 보니 DP로 풀면 되는 문제였음.
Rolling Hash라는걸 이용한 이분 탐색으로 풀어도 됨. 각 세트 자체를 해쉬로 만들어서 비교하는 방법인 듯.

------------------------------------------

DP로 푸는 방법은 생각보다 단순.
메모용 list를 만들고(m * n) 가장 마지막 인덱스부터 시작해서 같은 값이 있을 때마다 증분해주고, 그 중 최대값을 뽑아오면 됨.

class Solution:
    def findLength(self, nums1: List[int], nums2: List[int]) -> int:
        memo = [[0] * (len(nums2) + 1) for _ in range(len(nums1) + 1)]
        for i in range(len(nums1) - 1, -1, -1):
            for j in range(len(nums2) - 1, -1, -1):
                if nums1[i] == nums2[j]:
                    memo[i][j] = memo[i+1][j+1] + 1
        return max(max(row) for row in memo)
반응형