[Daily Leetcode] 575. Distribute Candies

Question:

Given an integer array with even length, where different numbers in this array represent different kinds of candies. Each number means one candy of the corresponding kind. You need to distribute these candies equally in number to brother and sister. Return the maximum number of kinds of candies the sister could gain.

Example 1:

Input: candies = [1,1,2,2,3,3]
Output: 3
Explanation:
There are three different kinds of candies (1, 2 and 3), and two candies for each kind.
Optimal distribution: The sister has candies [1,2,3] and the brother has candies [1,2,3], too. 
The sister has three different kinds of candies. 

Example 2:

Input: candies = [1,1,2,3]
Output: 2
Explanation: For example, the sister has candies [2,3] and the brother has candies [1,1]. 
The sister has two different kinds of candies, the brother has only one kind of candies. 

Note:

  1. The length of the given array is in range [2, 10,000], and will be even.
  2. The number in given array is in range [-100,000, 100,000].

Solution:

class Solution(object):
    def distributeCandies(self, candies):
        """
        :type candies: List[int]
        :rtype: int
        """

        if len(candies)/2 > len(set(candies)):
            return len(set(candies))
        else:
            return len(candies)/2

 

Analysis:

This question is very simple, and if you have read another post of my blog(http://liqichen.com/daily-leetcode500-keyboard-row/), the solution would come naturally and smoothly. Here it represent another usage of set() once you have figure out how to use this function in python many problems would be solved with easiness.
Another thing you have to notice in this solution is the algorithm which hides behind it: comparing the maximum number or the ‘baskets’ and the ‘candies’

Leave a Reply

Your email address will not be published. Required fields are marked *