Untitled
plain_text
a month ago
904 B
4
Indexable
Never
def findMinimumSum(rowA, rowB): num_zero_rowA = 0 num_zero_rowB = 0 sum_rowA = 0 sum_rowB = 0 for num in rowA: if num == 0: num_zero_rowA +=1 else: sum_rowA += num for num in rowB: if num == 0: num_zero_rowB +=1 else: sum_rowB += num if num_zero_rowA == 0 or num_zero_rowB == 0: return -1 if sum_rowA == sum_rowB: if num_zero_rowA == num_zero_rowB: # convert zeros to 1 for min sum return sum_rowA + num_zero_rowA else: if num_zero_rowA < num_zero_rowB: return sum_rowB + num_zero_rowB else: return sum_rowA + num_zero_rowA if sum_rowA < sum_rowB: return sum_rowB + num_zero_rowB else: return sum_rowA + num_zero_rowA if __name__ == '__main__': #print(findMinimumSum([1,0,2], [1,3,0,0])) print(findMinimumSum([2,5,0,1,1], [2,1,0,0]))