1452. People Whose List of Favorite Companies Is Not a Subset of Another List (Medium) - TengnanYao/daily_leetcode GitHub Wiki

class Solution:
    def peopleIndexes(self, favoriteCompanies: List[List[str]]) -> List[int]:
        h = {}
        n = len(favoriteCompanies)
        for i in range(n):
            for c in favoriteCompanies[i]:
                h[c] = h.get(c, []) + [i]
        result = []
        for i in range(n):
            arr = []
            for c in favoriteCompanies[i]:
                arr += h[c]
            counts = collections.Counter(arr)
            if len([v for v in counts.values() if v == len(favoriteCompanies[i])]) == 1:
                result.append(i)
        return result