LeetCode575-分糖果

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/distribute-candies/

中文链接：https://leetcode-cn.com/problems/distribute-candies/

题目详述

给定一个偶数长度的数组，其中不同的数字代表着不同种类的糖果，每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。

示例 1:

1
2
3
4
输入: candies = [1,1,2,2,3,3]
输出: 3
解析: 一共有三种种类的糖果，每一种都有两个。
     最优分配方案：妹妹获得[1,2,3],弟弟也获得[1,2,3]。这样使妹妹获得糖果的种类数最多。

示例 2 :

1
2
3
输入: candies = [1,1,2,3]
输出: 2
解析: 妹妹获得糖果[2,3],弟弟获得糖果[1,1]，妹妹有两种不同的糖果，弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。

注意:

  1. 数组的长度为[2, 10,000]，并且确定为偶数。
  2. 数组中数字的大小在范围[-100,000, 100,000]内。

题目详解

  • 因为要平均分配糖果，妹妹得到的糖果的数量为全部糖果数量的一半。
  • 如果糖果的种类数大于或等于糖果数量的一半，妹妹可以获得的糖果种类最多为糖果数量的一半。
  • 如果糖果的种类数小于糖果数量的一半，妹妹可以获得的糖果种类最多为糖果的种类数。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
public class LeetCode_00575 {

    public int distributeCandies(int[] candies) {
        int max = candies.length >> 1;
        Set<Integer> set = new HashSet<>();
        for (int candy : candies) {
            set.add(candy);
            if (set.size() >= max) {
                return max;
            }
        }
        return set.size();
    }
}

因为数组中数字的大小在范围[-100,000, 100,000]内，可以用一个大小为 200001 的数组来取代 HashSet。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public class LeetCode_00575 {

    public int distributeCandies(int[] candies) {
        int max = candies.length >> 1;
        boolean[] vis = new boolean[200001];
        int res = 0;
        for (int candy : candies) {
            if (!vis[candy + 100000]) {
                vis[candy + 100000] = true;
                if (++res == max) {
                    return max;
                }
            }
        }
        return res;
    }
}