Untitled

mail@pastecode.io avatarunknown
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]))