2021春季美团第九次笔试第一题

题目：

小团的蛋糕铺长期霸占着美团APP中“蛋糕奶茶”栏目的首位，因此总会吸引各路食客前来探店。

小团一天最多可以烤n个蛋糕，每个蛋糕有一个正整数的重量。

早上，糕点铺已经做好了m个蛋糕。

现在，有一个顾客要来买两个蛋糕，他希望买这一天糕点铺烤好的最重的和最轻的蛋糕，并且希望这两个蛋糕的重量恰好为a和b。剩余的n-m个蛋糕可以现烤，请问小团能否满足他的要求？

代码：

import java.io.*;
import java.io.IOException;
import java.util.TreeSet;
/**
 * @author zcy
 * @date 2021年04月03日 13:09
 */
public class Main {
    public static void main(String[] args) throws IOException{
        BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));
        String s;
        while((s=reader.readLine())!=null) {
            String[] str0 = s.split(" ");
            int n = Integer.parseInt(str0[0]);
            int m = Integer.parseInt(str0[1]);
            int a = Integer.parseInt(str0[2]);
            int b = Integer.parseInt(str0[3]);
            int t = Math.min(a, b);
            b = Math.max(a, b);
            a=t;
            String[] str1 = reader.readLine().split(" ");
            TreeSet<Integer> set=new TreeSet<>();
            for(String str2:str1){
                set.add(Integer.parseInt(str2));
            }
            int current_a=set.first();
            int current_b=set.last();
            if (current_a <a || current_b > b) {
                System.out.println("NO");
            }else{
            int count = 0;
            if (current_a == a) {
                count++;
            }
            if (current_b == b) {
                count++;
            }
            if ((count == 2) || (count == 1 && n - m >= 1) || (count == 0 && n - m >= 2)) {
                System.out.println("YES");
            } else {
                System.out.println("NO");
            }
            }
 
        }
    }
 
}