这道题作为B题还是比较善良的
这道题我们首先模拟一下过程,mex-p[i]=a[i],那我们求的p[i]就是mex-a[i],a数组我们已知,只用来找到每轮的mex就可以,那怎么找mex呢,显而易见,每轮mex是与上一轮的mex有关系的,那到底是正着看还是倒着看呢,这取决于哪个位置的mex最显而易见,很明显就是最后一个位置,最后的mex一定是n,所以p[ n ]=n-a[ n ],而再往前看,下一轮的mex, 是上面已有的p[ i ] 中最小值,所以很明显就想出来是倒序遍历,我们再构建一个set容器,为了维护有序,方便找到最小值
AC代码如下
#include<bits/stdc++.h>
using namespace std;
void solve(){
int n;
cin>>n;
vector<int>v(n+1);
vector<int>p(n+1);
set<int>st;
for(int i=1;i<=n;i++){
cin>>v[i];
}
p[n]=n-v[n];
st.insert(p[n]);
for(int i=n-1;i>0;i--){
p[i]=*st.begin()-v[i];
st.insert(p[i]);
}
for(int i=1;i<=n;i++){
cout<<p[i]<<" ";
}cout<<endl;
}
int main(){
int t;
cin>>t;
while(t--){
solve();
}
return 0;
}