제출 #1331637

#제출 시각아이디문제언어결과실행 시간메모리
1331637icebearTricks of the Trade (CEOI23_trade)C++20
50 / 100
983 ms177304 KiB
/* AUTHOR: TUAN ANH - BUI */
// ~~ icebear ~~
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> ii;
typedef pair<int, ii> iii;

template<class X, class Y>
    bool minimize(X &x, const Y &y) {
        if (x > y) return x = y, true;
        return false;
    }

template<class X, class Y>
    bool maximize(X &x, const Y &y) {
        if (x < y) return x = y, true;
        return false;
    }

#define FOR(i,a,b) for(int i=(a); i<=(b); ++i)
#define FORR(i,a,b) for(int i=(a); i>=(b); --i)
#define REP(i, n) for(int i=0; i<(n); ++i)
#define RED(i, n) for(int i=(n)-1; i>=0; --i)
#define MASK(i) (1LL << (i))
#define BIT(S, i) (((S) >> (i)) & 1)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(), x.end()
#define task "gen"
/*END OF TEMPLATE. ICEBEAR AND THE CAT WILL WIN TST26 */

const int MOD = 1e9 + 7;
const int inf = (int)1e9 + 27092008;
const ll INF  = (ll)1e18 + 27092008;
const int N = 250000 + 5;
int n, k, c[N], s[N];
ll pref[N];
int opt[N];

struct PersistentSegmentTree {
    struct Node {
        int l, r;
        int cnt;
        ll sum;
        Node(int _cnt = 0, ll _sum = 0): l(0), r(0), cnt(_cnt), sum(_sum) {}
    } node[N * 27];
    int version[N], curVer = 0, treeNode = 0;

    void combine(int id) {
        node[id].cnt = node[node[id].l].cnt + node[node[id].r].cnt;
        node[id].sum = node[node[id].l].sum + node[node[id].r].sum;
    }

    int build(int l, int r) {
        if (l == r) return ++treeNode;
        int mid = (l + r) >> 1;
        int id = ++treeNode;
        node[id].l = build(l, mid);
        node[id].r = build(mid + 1, r);
        combine(id);
        return id;
    }

    int update(int old, int l, int r, int pos, int val) {
        if (l == r) {
            node[++treeNode] = Node(node[old].cnt + 1, node[old].sum + val);
            return treeNode;
        }
        int mid = (l + r) >> 1;
        int id = ++treeNode;
        if (pos <= mid) {
            node[id].l = update(node[old].l, l, mid, pos, val);
            node[id].r = node[old].r;
        } else {
            node[id].l = node[old].l;
            node[id].r = update(node[old].r, mid + 1, r, pos, val);
        }
        combine(id);
        return id;
    }

    ll sumKth(int L, int R, int K, int l, int r) {
        if (K >= node[R].cnt - node[L].cnt) return node[R].sum - node[L].sum;
        if (l == r) return 0;
        int mid = (l + r) >> 1;
        if (node[node[R].r].cnt - node[node[L].r].cnt <= K)
            return node[node[R].r].sum - node[node[L].r].sum + sumKth(node[L].l, node[R].l, K - node[node[R].r].cnt + node[node[L].r].cnt, l, mid);
        else
            return sumKth(node[L].r, node[R].r, K, mid + 1, r);
    }

    void build() {
        version[0] = build(1, n);
    }

    void update(int pos, int val) {
        version[curVer + 1] = update(version[curVer], 1, n, pos, val);
        curVer++;
    }

    ll sumKth(int l, int r, int k) {
        return sumKth(version[l - 1], version[r], k, 1, n);
    }
} PerIT;

void DnC(int l, int r, int optL, int optR) {
    if (l > r) return;
    int mid = (l + r) >> 1;
    ll best = -INF;
    opt[mid] = n;
    FOR(i, max(optL, mid + k - 1), optR) {
        if (maximize(best, PerIT.sumKth(mid, i, k) - pref[i] + pref[mid - 1]))
            opt[mid] = i;
    }
    DnC(l, mid - 1, optL, opt[mid]);
    DnC(mid + 1, r, opt[mid], optR);
}

void init(void) {
    cin >> n >> k;
    FOR(i, 1, n) cin >> c[i], pref[i] = pref[i - 1] + c[i];
    FOR(i, 1, n) cin >> s[i];
    vector<ii> compress;
    FOR(i, 1, n) compress.pb(mp(s[i], i));
    sort(all(compress));
    vector<int> ord(n + 5, 0);
    FOR(i, 0, n - 1) ord[compress[i].se] = i + 1;
    PerIT.build();
    FOR(i, 1, n) PerIT.update(ord[i], s[i]);
}

void process(void) {
    DnC(1, n, 1, n);
    ll bestAll = -INF;
    vector<int> best;
    FOR(i, 1, n - k + 1) {
        ll val = PerIT.sumKth(i, opt[i], k) - pref[opt[i]] + pref[i - 1];
        if (maximize(bestAll, val)) best.clear();
        if (bestAll == val) best.pb(i);
    }

    cout << bestAll << '\n';

    best.pb(n); opt[n] = n;
    vector<ii> events;
    set<ii> ms;
    int l = 1, r = 0;
    ll sum = 0;
    REP(i, (int)best.size() - 1) {
        int L = best[i];
        int R = best[i + 1];
        while(r < opt[L] - 1) sum += s[++r], ms.insert(mp(s[r], r));
        while(l < L) {
            if (ms.find(mp(s[l], l)) != ms.end()) {
                sum -= s[l];
                ms.erase(mp(s[l], l));
            }
            l++;
        }

        while(r <= opt[R]) {
            sum += s[++r];
            ms.insert(mp(s[r], r));
            while (ms.size() > k) sum -= ms.begin()->fi, ms.erase(*ms.begin());
            if (sum - pref[r] + pref[l - 1] == bestAll && r - l + 1 >= k) {
                events.emplace_back(l, ms.begin()->fi);
                events.emplace_back(r + 1,- ms.begin()->fi);
            }
        }
    }

    sort(all(events));
    multiset<int> cur;
    int j = 0;

    FOR(i, 1, n) {
        while(j < (int)events.size() && events[j].fi <= i) {
            if (events[j].se < 0) cur.erase(cur.find(-events[j].se));
            else cur.insert(events[j].se);
            j++;
        }
        cout << (!cur.empty() && s[i] >= *cur.begin());
    }
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    if (fopen(task".inp", "r")) {
        freopen(task".inp", "r", stdin);
        freopen(task".out", "w", stdout);
    }
    int tc = 1;
//    cin >> tc;
    while(tc--) {
        init();
        process();
    }
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

trade.cpp: In function 'int main()':
trade.cpp:194:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  194 |         freopen(task".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
trade.cpp:195:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  195 |         freopen(task".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...