链接Problem - A - Codeforces
题目大意:n个非负整数数组a,MEX(a)=m,m是数组中缺失的最小数字(从0开始),例如1 0 3 4缺失2,0 2 2 0 缺失1,3 4 5 6缺失0,0 1 2 3 缺失4。现在你可以选择任意一个区间,将区间中所有数字修改为同一个值。问能否使得一次修改之后让MEX(a)=m+1。
分析:找到缺失数字X很容易,找到之后进行修改,使得数组不缺少X,而是缺少X+1。那么只需要把包含X+1的区间都修改成X。例如 0 2 2 0 缺失1,那么只要把所有的2都改成1,缺失变成2满足提议。
缺失X,此时数组a有几种情况:
(1)包含X+1。必须把X+1全部修改成X,因为是区间修改,所以修改后如样例3可能不满足要求。
(2)没有X+1,那么随便修改某个大于X的值为X就满足题意。比如 0 1 4 4,缺少2且没有3,随便找个大于2的改成2。
(3)特殊情况,什么时候绝对无法修改,如果n个数字缺失值为n,那么绝对无法修改得到n+1。
本题目属于基础性题目。
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios::sync_with_stdio(0);
int i,j,t,n,a[200005],b[200005];
cin>>t;
while(t--)
{
int meta=0,metb=0;
cin>>n;
for(i=1; i<=n; i++)
cin>>a[i],b[i]=a[i];
sort(a+1,a+n+1);
for(i=1; i<=n; i++)/**< 排个序找下第一个缺失数字meta */
if(a[i]==meta)
meta++;
else if(a[i]>meta)
break;
if(meta==n)/**< 缺失数字为n,那么无法构造 */
{
cout<<"NO\n";
continue;
}
i=1,j=n;/**< 缺失meta,目标修改一个区间缺失meta+1,所以要把所有meta+1都修改为meta */
while(i<=n&&b[i]!=meta+1)/**< 找到左侧第一个和右侧第一个meta+1 */
i++;
while(j>=1&&b[j]!=meta+1)
j--;
if(j<i)/**< 没有meta+1,那么随便找个没用数字改成meta+1即可 */
{
cout<<"YES\n";
continue;
}
for(; i<=j; i++)/**< 修改区间为meta */
b[i]=meta;
sort(b+1,b+n+1);
for(i=1; i<=n; i++)/**< 检查下新的缺失数字是不是meta+1 */
if(b[i]==metb)
metb++;
else if(b[i]>metb)
break;
if(meta+1==metb)
cout<<"YES\n";
else
cout<<"NO\n";
}
return 0;
}