C. Ehab and Prefix MEXs(模拟)

思路:首先判断是否有解,如果当前的a[i]>i,表示前面(i-1)个位置有i个数填好了,矛盾。之后,求解,先用集合将可能的选择存起来,如n=4,st=[0,1,2,3,4],然后扫一遍a[],把后序要出现的数预留好,也就是剔除一些当前位置不可以出现的数,stt=[0,0,0,2],st[1,3,4],当stt只剩下[2],表示0自由了,可以使用,加入到st。

int a[N];
int main() 
{
	//freopen("in.txt", "r", stdin);
	int n;
	scanf("%d", &n);
	int fg = 1;
	f(i, 1, n) { scanf("%d", &a[i]); if (a[i] > i)fg = 0; }
	if (!fg) { puts("-1"); }
	else
	{
		set<int> st;//can
		multiset<int> st2;//can't
		f(i, 0, n )st.insert(i);
		f(i, 1, n)
		{
			st2.insert(a[i]);
			if (st.count(a[i]))st.erase(st.find(a[i]));
		}
		f(i, 1, n)
		{
			auto it = st.begin();
			printf("%d ", *it);
			st.erase(it);
			auto itt=st2.find(a[i]);
			st2.erase(itt);
			if (st2.find(a[i]) == st2.end())st.insert(a[i]);
		}
	}
}

你可能感兴趣的:(Codeforces,pupil)