HTMLify

class Solution:
    def catchThieves(self, arr, k):
        n = len(arr)
        p = 0
        t = 0
        res = 0
        
        police = []
        thieves = []
        
        for i in range(n):
            if arr[i] == 'P':
                police.append(i)
            else:
                thieves.append(i)
                
        while p < len(police) and t < len(thieves):
            if abs(police[p] - thieves[t]) <= k:
                res += 1
                p += 1
                t += 1
            elif police[p] < thieves[t]:
                p += 1
            else:
                t += 1
                
        return res