Submission #1156184

#	Time	Username	Problem	Language	Result	Execution time	Memory
1156184		guagua0407	Fish 3 (JOI24_fish3)	C++20	100 / 100	330 ms	86872 KiB

Main

//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define f first
#define s second
#define all(x) x.begin(),x.end()
#define _ ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);

int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};

void setIO(string s) {
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}

const ll inf=(ll)1e18;

int main() {_
    int n;
    ll d;
    cin>>n>>d;
    vector<ll> c(n);
    for(int i=0;i<n;i++){
        cin>>c[i];
    }
    vector<vector<pair<int,int>>> qs(n);
    int q;
    cin>>q;
    for(int i=0;i<q;i++){
        int l,r;
        cin>>l>>r;
        l--;
        r--;
        qs[r].push_back({l,i});
    }
    vector<ll> a(n);
    for(int i=n-2;i>=0;i--){
        a[i]=max(0ll,(c[i]-c[i+1]+d-1)/d);
        c[i]-=a[i]*d;
    }
    vector<ll> pre(n);
    pre[0]=a[0];
    for(int i=1;i<n;i++){
        pre[i]=pre[i-1]+a[i];
    }
    vector<vector<ll>> mn(n,vector<ll>(20,-inf));
    for(int i=0;i<n;i++){
        mn[i][0]=c[i];
    }
    for(int j=1;j<20;j++){
        for(int i=0;i+(1<<j)<=n;i++){
            mn[i][j]=min(mn[i][j-1],mn[i+(1<<(j-1))][j-1]);
        }
    }
    auto get=[&](int l,int r){
        int bit=__lg(r-l+1);
        return min(mn[l][bit],mn[r-(1<<bit)+1][bit]);
    };
    vector<pair<pair<int,int>,pair<ll,ll>>> st;
    st.push_back({{-1,0},{-1,0}});
    vector<ll> ans(q);
    for(int i=0;i<n;i++){
        while(st.back().s.f>=a[i]){
            st.pop_back();
        }
        int tf=((get(st.back().f.f+1,i)+a[i]*d)<0);
        st.push_back({{i,st.back().f.s+tf},{a[i],st.back().s.s+a[i]*(i-st.back().f.f)}});
        for(auto v:qs[i]){
            auto [l,id]=v;
            int pos=lower_bound(all(st),make_pair(make_pair(l,-1),make_pair(-1ll,-1ll)))-st.begin();
            int bad=st.back().f.s-st[pos].f.s;
            if(bad>0){
                ans[id]=-1;
                continue;
            }
            int bad2=((get(l,st[pos].f.f)+st[pos].s.f*d)<0);
            if(bad2){
                ans[id]=-1;
                continue;
            }
            ans[id]=pre[i]-(l==0?0:pre[l-1]);
            ans[id]-=(st.back().s.s-st[pos].s.s+st[pos].s.f*(st[pos].f.f-l+1));
        }
    }
    for(int i=0;i<q;i++){
        cout<<ans[i]<<'\n';
    }
    return 0;
}
//maybe its multiset not set
//yeeorz
//diaoborz

Compilation message (stderr)

Main.cpp: In function 'void setIO(std::string)':
Main.cpp:15:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   15 |     freopen((s + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:16:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   16 |     freopen((s + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 9 / 9
• Subtask 2: 7 / 7
• Subtask 3: 28 / 28
• Subtask 4: 20 / 20
• Subtask 5: 36 / 36