题目内容

小红正在研究数组的$mex$。她有一个长度为n的数组{$a_1,a_2,...,a_n$},她想知道$mex${$a$}是否为$k$？如果是的话,直接输出$YES$;否则，小红可以从下方两个操作中选样一个执行一次:

• 选择一个元素$a_i$加一,使得$a_i$变为$a_i+1$;
• 选择一个元素$a_i(a_i>0)$减一,使得$a_i$变为$a_i-1$

题解

题面描述

给定一个长度为 $n$ 的数组 $\{a_1,a_2,\dots,a_n\}$，定义数组的 $\mathrm{mex}$ 为最小的不在数组中的非负整数。例如

$\mathrm{mex}\{1,2,3\}=0$,$\quad \mathrm{mex}\{0,2,5\}=1.$

现在给定一个目标值 $k$，小红想知道数组的 $\mathrm{mex}$ 是否已经等于 $k$？如果是，输出 $\texttt{YES}$；否则，她允许对数组执行 恰好一次 操作，操作为：