1. 程式人生 > >【LeetCode】分糖果

【LeetCode】分糖果

給定一個偶數長度的陣列,其中不同的數字代表著不同種類的糖果,每一個數字代表一個糖果。你需要把這些糖果平均分給一個弟弟和一個妹妹。返回妹妹可以獲得的最大糖果的種類數。

示例 1:

輸入: candies = [1,1,2,2,3,3]
輸出: 3
解析: 一共有三種種類的糖果,每一種都有兩個。
     最優分配方案:妹妹獲得[1,2,3],弟弟也獲得[1,2,3]。這樣使妹妹獲得糖果的種類數最多。

示例 2 :

輸入: candies = [1,1,2,3]
輸出: 2
解析: 妹妹獲得糖果[2,3],弟弟獲得糖果[1,1],妹妹有兩種不同的糖果,弟弟只有一種。這樣使得妹妹可以獲得的糖果種類數最多。

注意:

  1. 陣列的長度為[2, 10,000],並且確定為偶數。
  2. 陣列中數字的大小在範圍[-100,000, 100,000]內。

import java.util.*;

public class Main {

    public static void main(String[] args)
    {

        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();
        int arr[]=new int[n];
        for (int i=0;i<arr.length;i++)
            arr[i]=sc.nextInt();

        System.out.println(Solution.distributeCandies(arr));

    }

}


class Solution {
    public static int distributeCandies(int[] candies) {

        int candiesNum = 0;         //總的糖果種類數
        Set set = new HashSet();      //set不儲存重複的元素。
        for (int i = 0; i < candies.length; i++)
            set.add(candies[i]);    
        candiesNum = set.size();    //找出不重複的糖果種類數

        /*如果妹妹最大能得到的糖果數小於等於總的糖果種類數:
        則她能得到的最大糖果種類數為她所得的糖果總數
        反之即為糖果的最大種類數*/
        if (candies.length / 2 <= candiesNum)    
            return candies.length / 2;
        return candiesNum;
    }
}

思路:如果妹妹最大能得到的糖果數小於等於總的糖果種類數:則她能得到的最大糖果種類數為她所得的糖果總數
反之即為糖果的最大種類數