题目传送门
题目大意
给你一个长度为n的数组a,每个数字对应一个0或者1,为0即为可以移动,你可以任意移动对应为0的数字
使得前缀和数组p的最大值所在的位置最小
输出移动后的数组
思路
为使得前缀和的最大值尽可能的早出现,所以我们需要将可以移动的值中大的值尽可能的往前放
即为将可以移动的位置的值按非递增的方式培训然后插入不可移动中
AC Code
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
#define endl '\n'
#define INF 0x3f3f3f3f
#define int long long
#define debug(a) cout<<#a<<"="<<a<<endl;
// #define TDS_ACM_LOCAL
const int N=2e5 +9;
int n, a[N], flag[N];
int b[N];
void solve(){
cin>>n;
memset(b,0,sizeof(b));
int k=0;
for(int i=1; i<=n; i++) cin>>a[i];
for(int i=1; i<=n; i++){
cin>>flag[i];
if(flag[i]==0){
b[++k]=a[i];
}
}
sort(b+1, b+1+k);
for(int i=1; i<=n; i++){
if(flag[i]==1) cout<<a[i]<<" ";
else cout<<b[k--]<<" ";
}
cout<<endl;
return ;
}
signed main(){
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
#ifdef TDS_ACM_LOCAL
freopen("D:\\VS code\\.vscode\\testall\\in.txt", "r", stdin);
freopen("D:\\VS code\\.vscode\\testall\\out.txt", "w", stdout);
#endif
int T;
cin>>T;
while(T--) solve();
return 0;
}