Submission #700334

#	Time	Username	Problem	Language	Result	Execution time	Memory
700334		LeVanThuc	Discharging (NOI20_discharging)	C++17	47 / 100	1051 ms	182212 KiB

Discharging

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define fi first
#define se second
#define p(x,y) pair<ll,ll>(x,y)
#define BIT(i,x) ((x>>i)&1)
#define MASK(x) (1<<x)
#define ld long double
#define __builtin_popcount __builtin_popcountll
#define pll pair<ll,ll>
template<class T1,class T2>
bool maximize(T1 &x,const T2 &y)
{
    if(x<y)
    {
        x=y;
        return 1;
    }
    return 0;
}
template<class T1,class T2>
bool minimize(T1 &x,const T2 &y)
{
    if(x>y)
    {
        x=y;
        return 1;
    }
    return 0;
}
void online()
{
    std::ios_base::sync_with_stdio(0);
    cin.tie(0);
#ifndef ONLINE_JUDGE

#else
#endif
}
static const ll N=1e6+10,M=1e16;
ll n,m,A,B,a[N][21],b[N],f[N],po[21];
ll rmq(ll l,ll r)
{
    ll z=log2(r-l);
    return max(a[l][z],a[r-po[z]+1][z]);
}
ll c(ll l,ll r)
{
    return rmq(l,r)*(n-l+1);
}
int main()
{
    online();
    cin>>n;
    for(ll i=1;i<=n;i++)
    {
        cin>>a[i][0];
    }
    po[0]=1;
    for(int k=1;k<=19;k++) po[k]=po[k-1]*2;
    for(int j=1;j<=19;j++)
    for(int i=1;i+po[j]-1<=n;i++)
    {
        a[i][j]=max(a[i][j-1],a[i+po[j-1]][j-1]);
    }
    f[0]=0;
    vector<pll> vt;
    vt.emplace_back(0,0);
    ll ans=M;
    for(ll i=1;i<=n;i++)
    {
        auto best=lower_bound(vt.begin(),vt.end(),p(i+1,0));
        best--;
        auto x=*best;
        ll k=x.se;
        f[i]=f[k]+c(k+1,i);
        for(ll j=vt.size()-1;j>=0;j--)
        {
            if(f[vt[j].se]+c(vt[j].se+1,vt[j].fi)>f[i]+c(i+1,vt[j].fi))
            {
                vt.pop_back();
            }
            else
            {
                ll b=vt[j].fi,d=n/2+1;
                 while(d)
                {
                    while(b+d<=n&&f[vt[j].se]+c(vt[j].se+1,b+d)<=f[i]+c(i+1,b+d)) b+=d;
                    d/=2;
                }
                b++;
                if(b!=n+1)
                {
                    vt.emplace_back(b,i);
                }
                break;
            }
        }
        if(vt.empty()) vt.emplace_back(1,i);
//        for(auto v:vt) cout<<"("<<v.fi<<","<<v.se<<") ";
//        cout<<f[i];
//        cout<<'\n';
    }
    cout<<f[n];
}

Compilation message (stderr)

Discharging.cpp: In function 'int main()':
Discharging.cpp:70:8: warning: unused variable 'ans' [-Wunused-variable]
   70 |     ll ans=M;
      |        ^~~

• Subtask 1: 9 / 9
• Subtask 2: 13 / 13
• Subtask 3: 0 / 25
• Subtask 4: 11 / 11
• Subtask 5: 14 / 14
• Subtask 6: 0 / 28