[LeetCode] 1704. Determine if String Halves Are Alike

1704. Determine if String Halves Are Alike

Easy

You are given a string s of even length. Split this string into two halves of equal lengths, and let a be the first half and b be the second half.

Two strings are alike if they have the same number of vowels ('a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U'). Notice that s contains uppercase and lowercase letters.

Return true if a and b are alike. Otherwise, return false.

 

Example 1:

Input: s = "book"
Output: true
Explanation: a = "bo" and b = "ok". a has 1 vowel and b has 1 vowel. Therefore, they are alike.

Example 2:

Input: s = "textbook"
Output: false
Explanation: a = "text" and b = "book". a has 1 vowel whereas b has 2. Therefore, they are not alike.
Notice that the vowel o is counted twice.

 

Constraints:

• 2 <= s.length <= 1000
• s.length is even.
• s consists of uppercase and lowercase letters.

---

문제 풀이

• 주어진 문장은 짝수의 길이를 가지고 있다. 
• 이 문장을 반으로 나눴을때 대소문자 관계없이 모음(Vowel)의 개수가 같으면 True 아니면 False를 리턴한다.
• 문제 제한을 보면 문장의 최대길이가 1000이며, O(N^2)로도 풀린다.
• 문장전체를 소문자로 바꾼다.
• Counter를 사용해 반을 나눈 문장의 알파벳의 개수를 센다.
• 반으로 나뉜 두 문장에서 모음의 개수를 세어 답을 리턴한다.

소스 코드

class Solution:
    def halvesAreAlike(self, s: str) -> bool:
        s = s.lower()
        half_len = len(s)//2
        s1 = Counter(s[:half_len])
        s2 = Counter(s[half_len:])
        total1, total2 = 0, 0
        for l in ('a', 'e', 'i', 'o', 'u'):
            total1 += s1[l]
            total2 += s2[l]
        return total1 == total2