[LeetCode] 3042. Count Prefix and Suffix Pairs

https://leetcode.com/problems/count-prefix-and-suffix-pairs-i

Count Prefix and Suffix Pairs I - LeetCode

주어진 string 리스트에서 하나의 단어가 다른 단어의 prefix 와 suffix 인 경우를 세는 문제이다.

isPrefixAndSuffix 라는 함수를 구현해서 하나의 단어가 다른 단어의 prefix와 suffix를 모두 만족하는 경우에는 True, 아닌 경우에는 False를 리턴하면 된다.

다만 문제를 잘 읽어보면 ,
Returnan integer denoting the number of index pairs (i, j) such that i < j, and isPrefixAndSuffix(words[i], words[j]) is true. 라고되어 있기 때문에 i는 주어진 리스트 길이의 -1 까지만 순회를 해야하며 j는 i 보다는 항상 커야 한다. (i+1 부터 순회)

class Solution:
    def countPrefixSuffixPairs(self, words: List[str]) -> int:
        answer = 0
        for i in range(len(words)-1):
            for j in range(i+1, len(words)):
                if isPrefixAndSuffix(words[i], words[j])==True:
                    answer += 1
        return answer

    
def isPrefixAndSuffix(str1, str2):
    if str2.startswith(str1) and str2.endswith(str1):
        return True
    else:
        return False

accept!