CF - E95(div2) -- B. Negative Prefixes【贪心】

题意

给定一个序列，然后给定那些位置被锁定不能被移动，将序列可移动的数随机移动，移动后为a1, a2, a3…an;
设p1 = a1 ，p2 = a1 + a2 ，p为a的前缀和，k为前缀和中pi < 0 的个数
问怎样移动使得k最小。

思路

让可移动的数尽量把最大的数放前面，使得前面的前缀和为正数。这样会使得k最小。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int a[maxn], b[maxn], c[maxn];
void solve() {
    int n, cnt = 0;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
    }
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &b[i]);
        if (b[i] == 0) {
            c[cnt++] = a[i];
        }
    }
    sort(c, c + cnt);
    for (int i = 1; i <= n; ++i) {
        if (b[i] == 0) {
            a[i] = c[--cnt];
        }
    }
    for (int i = 1; i <= n; ++i) {
        if (i < n) {
            printf("%d ", a[i]);
        } else {
            printf("%d\n", a[i]);
        }
    }
}
int main() {
    int t;
    scanf("%d", &t);
    for (int i = 0; i < t; ++i) {
        solve();
    }
    return 0;
}