575.分糖果

【LetMeFly】575.分糖果：min(type, size/2)

力扣题目链接：https://leetcode.cn/problems/distribute-candies/

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

解题方法：比较

限制Alice能吃到糖的种类的因素有两个：

1. 糖本身的种类——无论Alice使用什么策略都无法突破糖原本种类数的限制；
2. 糖的总个数——医生让她最多吃一半数量的糖。

因此最终答案为$\min(type, \frac{size}2)$

• 时间复杂度$O(size)$
• 空间复杂度$O(size)$

AC代码

C++

class Solution {
public:
    int distributeCandies(vector<int>& candyType) {
        set<int> se(candyType.begin(), candyType.end());
        return min(se.size(), candyType.size() / 2);
    }
};

Go

package main

func min(a int, b int) int {
    if a <= b {
        return a
    }
    return b
}

func distributeCandies(candyType []int) int {
    se := make(map[int]int)
    for _, t := range candyType {
        se[t] = 0
    }
    return min(len(se), len(candyType) / 2)
}

Java

// import java.util.HashSet;
// import java.util.Set;

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

Python

# from typing import List

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

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/139387726