[LeetCode] 1641 | Count Sorted Vowel Strings | Python

---

문제 링크: https://leetcode.com/problems/count-sorted-vowel-strings/

---

문제

Given an integer n, return the number of strings of length n that consist only of vowels (a, e, i, o, u) and are lexicographically sorted.

A string s is lexicographically sorted if for all valid i, s[i] is the same as or comes before s[i+1] in the alphabet.

---

조건

• 1 <= n <= 50 

---

내 풀이

순서는 유지하면서 중복은 허용하는 순열의 수를 센다.

itertools의 combinations_with_replacement를 사용하면 바로 구할 수 있다.

---

코드

from itertools import combinations_with_replacement

class Solution:
    def countVowelStrings(self, n: int) -> int:
        return len(list(combinations_with_replacement([x for x in range(5)], n)))