답안 #816919

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
816919 2023-08-09T07:29:39 Z Jarif_Rahman 모임들 (IOI18_meetings) C++17
36 / 100
774 ms 5244 KB
#include "meetings.h"
#include <bits/stdc++.h>
#define pb push_back
#define f first
#define sc second
using namespace std;
typedef long long int ll;
typedef string str;

struct segtree{
    int k;
    vector<int> mx;
    segtree(int n){
        k = 1;
        while(k < n) k*=2;
        mx.assign(2*k, 0);
    }
    void update(int i, int x){
        i+=k;
        mx[i] = x;
        i/=2;
        while(i){
            mx[i] = max(mx[2*i], mx[2*i+1]);
            i/=2;
        }
    }
    int get(int l, int r, int nd, int a, int b){
        if(a > r || b < l) return 0;
        if(a >= l && b <= r) return mx[nd];
        int md = (a+b)/2;
        return max(get(l, r, 2*nd, a, md), get(l, r, 2*nd+1, md+1, b));
    }
    int get(int l, int r){
        return get(l, r, 1, 0, k-1);
    }
};

vector<ll> minimum_costs_subtask_nq(const vector<int>& _H, const vector<int>& L, const vector<int>& R){
    vector<ll> H;
    int n = _H.size(), q = L.size();
    for(int i = 0; i < n; i++) H.pb(_H[i]);

    vector<ll> ans(q);
    for(int qi = 0; qi < q; qi++){
        int l = L[qi], r = R[qi];
        vector<ll> costs(r-l+1, 0);

        stack<pair<ll, int>> st;
        ll c = 0;
        for(int i = l; i <= r; i++){
            while(!st.empty() && st.top().f <= H[i]){
                auto [h, j] = st.top();
                st.pop();
                if(st.empty()) c-=h*(j-l+1);
                else c-=h*(j-st.top().sc);
            }
            if(st.empty()) c+=H[i]*(i-l+1);
            else c+=H[i]*(i-st.top().sc);
            st.push({H[i], i});
            costs[i-l]+=c;
        }
        while(!st.empty()) st.pop();
        c = 0;
        for(int i = r; i >= l; i--){
            while(!st.empty() && st.top().f <= H[i]){
                auto [h, j] = st.top();
                st.pop();
                if(st.empty()) c-=h*(r-j+1);
                else c-=h*(st.top().sc-j);
            }
            if(st.empty()) c+=H[i]*(r-i+1);
            else c+=H[i]*(st.top().sc-i);
            st.push({H[i], i});
            costs[i-l]+=c;
            costs[i-l]-=H[i];
        }

        ans[qi] = *min_element(costs.begin(), costs.end());
    }

    return ans;
}

vector<ll> minimum_costs(vector<int> H, vector<int> L, vector<int> R){
    int n = H.size(), q = L.size();
    if(n <= 5000 && q <= 5000) return minimum_costs_subtask_nq(H, L, R);

    vector<int> left(n, 0), right(n, 0);
    segtree seg(n);
    for(int i = 0; i < n; i++) if(H[i] == 1) {
        left[i] = 1;
        if(i) left[i]+=left[i-1];
        seg.update(i, left[i]);
    }
    for(int i = n-1; i >= 0; i--) if(H[i] == 1){
        right[i] = 1;
        if(i+1 < n) right[i]+=right[i+1];
    }

    vector<ll> ans(q, 0);
    for(int qi = 0; qi < q; qi++){
        int l = L[qi], r = R[qi];

        if(right[l] >= r-l+1){
            ans[qi] = r-l+1;
            continue;
        }

        int a = right[l], b = left[r];
        ans[qi] = 2*(r-l+1)-max(max(a, b), seg.get(l+a, r-b));
    }

    return ans;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 2 ms 312 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 2 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 2 ms 312 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 2 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 242 ms 660 KB Output is correct
11 Correct 764 ms 648 KB Output is correct
12 Correct 238 ms 656 KB Output is correct
13 Correct 774 ms 652 KB Output is correct
14 Correct 199 ms 648 KB Output is correct
15 Correct 196 ms 660 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 21 ms 1652 KB Output is correct
3 Correct 70 ms 5244 KB Output is correct
4 Correct 57 ms 5192 KB Output is correct
5 Correct 40 ms 5180 KB Output is correct
6 Correct 46 ms 5180 KB Output is correct
7 Correct 50 ms 5176 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 21 ms 1652 KB Output is correct
3 Correct 70 ms 5244 KB Output is correct
4 Correct 57 ms 5192 KB Output is correct
5 Correct 40 ms 5180 KB Output is correct
6 Correct 46 ms 5180 KB Output is correct
7 Correct 50 ms 5176 KB Output is correct
8 Incorrect 58 ms 5188 KB Output isn't correct
9 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 2 ms 312 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 2 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 242 ms 660 KB Output is correct
11 Correct 764 ms 648 KB Output is correct
12 Correct 238 ms 656 KB Output is correct
13 Correct 774 ms 652 KB Output is correct
14 Correct 199 ms 648 KB Output is correct
15 Correct 196 ms 660 KB Output is correct
16 Correct 0 ms 212 KB Output is correct
17 Correct 21 ms 1652 KB Output is correct
18 Correct 70 ms 5244 KB Output is correct
19 Correct 57 ms 5192 KB Output is correct
20 Correct 40 ms 5180 KB Output is correct
21 Correct 46 ms 5180 KB Output is correct
22 Correct 50 ms 5176 KB Output is correct
23 Incorrect 58 ms 5188 KB Output isn't correct
24 Halted 0 ms 0 KB -