题解：CF1418B Negative Prefixes

Wide_Master · 2025-01-25 11:48:29 · 题解

分析

考虑贪心。我们将所有没被上锁的数字提取出来，然后从大到小排序，在放回原数组即可。

明显发现，如果想要让 k 较大，那么位置越靠前的对整体的贡献越大，所以贪心策略无误。

注意多测清空。

代码

//By xiaozhou001
#include<bits/stdc++.h>
using namespace std;
const int N=3e5+7;
int t,n,a[N],b[N],c[N],cnt;
int main()
{
    cin>>t;
    while(t--){
        cin>>n;
        cnt=0;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        for(int i=1;i<=n;i++){
            cin>>b[i];
            if(!b[i])c[++cnt]=a[i];
        }
        sort(c+1,c+1+cnt,greater<int>());
        cnt=0;
        for(int i=1;i<=n;i++){
            if(b[i])cout<<a[i]<<" ";
            else cout<<c[++cnt]<<" ";
        }
        cout<<endl;
    }
    return 0;
}