Check Array Formation Through Concatenation

001 check arry.py

class Solution(object):
    def canFormArray(self, arr, pieces):
        """
        :type arr: List[int]
        :type pieces: List[List[int]]
        :rtype: bool
        """
        mp = {x[0]: x for x in pieces}
        res = []
        
        for num in arr:
            res += mp.get(num, [])
        
        return res == arr

002 Count Vowel.py

class Solution(object):
    def countVowelStrings(self, n: int) -> int:
        res = [1] * 5
        for _ in range(n-1):
            for i in range(1, 5):
                res[i] += res[i-1]
        return sum(res)