Submission #446103

#	Time	Username	Problem	Language	Result	Execution time	Memory
446103		Ronin13	Fountain (eJOI20_fountain)	C++14	100 / 100	309 ms	24656 KiB

fountain

#include<bits/stdc++.h>
#define ll long long
#define f first
#define s second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define ull unsigned ll
#define epb emplace_back
#define pb push_back
#define inf 1e9+1
#define linf 1e18+1
using namespace std;

void solve(){
    int n;cin>>n;
    int q;cin>>q;
    int  c[n+2],d[n+2];
    for(int i=1;i<=n;i++){
        cin>>d[i]>>c[i];
    }
    stack<int>st;
    st.push(n+1);
    d[n+1]=2*inf+1;
    int next[n+2][22];
    int water[n+2][22];
    next[n+1][0]=n+1;
    int par[n+2];
    for(int i=1;i<=n+1;i++){
        par[i]=0;
    }
    int pref[n+2];
    pref[0]=0;
    pref[n+1]=2*inf;

    for(int i=n;i>=1;i--){
        while(true){
            int ind=st.top();
            if(d[ind]>d[i]){
                next[i][0]=ind;
                water[i][0]=c[i];
                par[ind]=i;
                break;
            }
            else st.pop();
        }
        st.push(i);
    }
    for(int j=1;j<=21;j++){
        for(int i=1;i<=n+1;i++){
            next[i][j]=next[next[i][j-1]][j-1];
            water[i][j]=water[i][j-1]+water[next[i][j-1]][j-1];
        }
    }


    while(q--){
        int ves;int l;
        cin>>ves>>l;

        for(int i=21;i>=0;i--){
            int nex=next[ves][i];
            if(l>water[ves][i])l-=water[ves][i],ves=nex;
        }
        cout<<ves%(n+1)<<"\n";
    }
}

int main(){
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    solve();
}

Compilation message (stderr)

fountain.cpp: In function 'void solve()':
fountain.cpp:27:9: warning: variable 'par' set but not used [-Wunused-but-set-variable]
   27 |     int par[n+2];
      |         ^~~
fountain.cpp:31:9: warning: variable 'pref' set but not used [-Wunused-but-set-variable]
   31 |     int pref[n+2];
      |         ^~~~

• Subtask 1: 30 / 30
• Subtask 2: 30 / 30
• Subtask 3: 40 / 40