2023.11.29 Today's Challenge

fabichoi 2023. 11. 29. 23:45

https://leetcode.com/problems/number-of-ways-to-divide-a-long-corridor/

Number of Ways to Divide a Long Corridor - LeetCode

Can you solve this real interview question? Number of Ways to Divide a Long Corridor - Along a long library corridor, there is a line of seats and decorative plants. You are given a 0-indexed string corridor of length n consisting of letters 'S' and 'P' wh

leetcode.com

DP로 푸는 문제. 변수선언을 하는 특이한 방법에 대해서 배움 (1_000_000_000)

class Solution:
    def numberOfWays(self, corridor: str) -> int:
        MOD = 1_000_000_007
        cache = [[-1] * 3 for _ in range(len(corridor))]

        def count(index, seats):
            if index == len(corridor):
                return 1 if seats == 2 else 0
            if cache[index][seats] != -1:
                return cache[index][seats]
            if seats == 2:
                if corridor[index] == "S":
                    result = count(index + 1, 1)
                else:
                    result = (count(index+1, 0) + count(index+1, 2)) % MOD
            else:
                if corridor[index] == "S":
                    result = count(index+1, seats+1)
                else:
                    result = count(index+1, seats)            
            cache[index][seats] = result
            return cache[index][seats]        
        return count(0, 0)