D. Winter Is Coming 贪心（好题）

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <assert.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;


#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
const int maxn = 2e5 + 20;
int a[maxn];
struct node {
    int val, id;
    node(int vval, int iid) : val(vval), id(iid) {}
    bool operator < (const struct node & rhs) const {
        if (val != rhs.val) return val < rhs.val;
        else return id < rhs.id;
    }
};
vector<int>pos;
vector<int>dis;
set<struct node>ss;
bool vis[maxn];
void work() {
    int n, k;
    cin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
        if (a[i] < 0) pos.push_back(i);
    }
    if (k < pos.size()) {
        cout << -1 << endl;
        return;
    }
    if (pos.size() == 0) {
        cout << 0 << endl;
        return;
    }
    for (int i = 1; i < pos.size(); ++i) {
        dis.push_back(pos[i] - pos[i - 1] - 1);
        ss.insert(node(pos[i] - pos[i - 1] - 1, i - 1));
    }
    int lef = k - pos.size();
    for (set<struct node> :: iterator it = ss.begin(); it != ss.end(); ++it) {
//        cout << it->val << endl;
        if (lef >= it->val) {
//            cout << "ff" << endl;
            lef -= it->val;
            vis[it->id] = true;
        } else break;
    }
    int ans = 1;
    for (int i = 0; i < dis.size(); ++i) {
        if (vis[i]) continue;
        ans += 2;
    }
    if (lef < n - pos.back()) {
        ans += 1;
    }
    cout << ans << endl;
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    work();
    return 0;
}