Hard
Maximum Good People Based on Statements — Python
Full explanation · Time O(n^2 * 2^n) · Space O(1)
# Time: O(n^2 * 2^n)
# Space: O(1)
# brute force, bitmask
class Solution(object):
def maximumGood(self, statements):
"""
:type statements: List[List[int]]
:rtype: int
"""
def check(mask):
return all(((mask>>j)&1) == statements[i][j]
for i in xrange(len(statements)) if (mask>>i)&1
for j in xrange(len(statements[i])) if statements[i][j] != 2)
def popcount(x):
result = 0
while x:
x &= x-1
result += 1
return result
result = 0
for mask in xrange(1<<len(statements)):
if check(mask):
result = max(result, popcount(mask))
return result