当前位置: 代码迷 >> 综合 >> B. Negative Prefixes(贪心,构造)
  详细解决方案

B. Negative Prefixes(贪心,构造)

热度:2   发布时间:2024-02-24 13:58:48.0
题目传送门

Negative Prefixes

题目大意

给你一个长度为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_LOCALfreopen("D:\\VS code\\.vscode\\testall\\in.txt", "r", stdin);freopen("D:\\VS code\\.vscode\\testall\\out.txt", "w", stdout);
#endifint T;cin>>T;while(T--)  solve();return 0;
}
  相关解决方案