Question

575. 分糖果

English Version

题目描述

Alice 有 n 枚糖，其中第 i 枚糖的类型为 candyType[i] 。Alice 注意到她的体重正在增长，所以前去拜访了一位医生。

医生建议 Alice 要少摄入糖分，只吃掉她所有糖的 n / 2 即可（n 是一个偶数）。Alice 非常喜欢这些糖，她想要在遵循医生建议的情况下，尽可能吃到最多不同种类的糖。

给你一个长度为 n 的整数数组 candyType ，返回： Alice _在仅吃掉 n / 2 枚糖的情况下，可以吃到糖的 最多 种类数_。

 

示例 1：

输入：candyType = [1,1,2,2,3,3]
输出：3
解释：Alice 只能吃 6 / 2 = 3 枚糖，由于只有 3 种糖，她可以每种吃一枚。

示例 2：

输入：candyType = [1,1,2,3]
输出：2
解释：Alice 只能吃 4 / 2 = 2 枚糖，不管她选择吃的种类是 [1,2]、[1,3] 还是 [2,3]，她只能吃到两种不同类的糖。

示例 3：

输入：candyType = [6,6,6,6]
输出：1
解释：Alice 只能吃 4 / 2 = 2 枚糖，尽管她能吃 2 枚，但只能吃到 1 种糖。

 

提示：

• n == candyType.length
• 2 <= n <= 104
• n 是一个偶数
• -105 <= candyType[i] <= 105

解法

方法一

class Solution:
  def distributeCandies(self, candyType: List[int]) -> int:
    return min(len(candyType) >> 1, len(set(candyType)))

class Solution {
  public int distributeCandies(int[] candyType) {
    Set<Integer> s = new HashSet<>();
    for (int c : candyType) {
      s.add(c);
    }
    return Math.min(candyType.length >> 1, s.size());
  }
}

class Solution {
public:
  int distributeCandies(vector<int>& candyType) {
    unordered_set<int> s;
    for (int c : candyType) s.insert(c);
    return min(candyType.size() >> 1, s.size());
  }
};

func distributeCandies(candyType []int) int {
  s := hashset.New()
  for _, c := range candyType {
    s.Add(c)
  }
  return min(len(candyType)>>1, s.Size())
}