코딩 공부/Leetcode

[Python] 3042. Count Prefix and Suffix Pairs I

일하는 공학도 2025. 1. 8. 16:00
728x90

 

  1. words에 나온 요소들로 (i, j) pair를 만드는데, i < j여야 한다.
  2. isPrefixAndSuffix(str1, str2) 함수는, str1의 문자가 str2의 '접두사'와 '접미사' 둘 다 사용되었는가의 여부를 확인한다.
  3. words에서 만든 (i, j) pair들 중 isPrefixAndSuffix 함수를 만족하는 갯수를 return
class Solution(object):
    def countPrefixSuffixPairs(self, words):
        """
        :type words: List[str]
        :rtype: int
        """
        n = len(words)
        ans = 0
        for i in range(n):
            a = len(words[i])
            for j in range(i+1,n):
                if words[j][:a] == words[i] and words[j][(-a):] == words[i]:
                    ans += 1
        return ans
  1. n으로 words의 요소가 몇 개인지 지정
  2. a는 words 내부의 문자가 몇 개인지 확인
  3. j는 range(i+1, n)으로 지정하여, i < j를 구현
  4. words[j][:a]는 접두사, words[j][(-a):]는 접미사를 의미하고, words[i]와 동일한지 확인

Runtime : 3ms (100%)

Memory : 12.27MB (43%)

728x90