481. Magical String

Question

CleanShot 2024-12-05 at 20.29.02.png

Code

class Solution:
    def magicalString(self, n: int) -> int:
        
        s = ['1', '2', '2']

        for i in range(2, n):

            if s[-1] == '2':
                s += ['1'] * int(s[i])
            else:
                s += ['2'] * int(s[i])
            
            if (len(s) > n): break

        return s[:n].count('1')