제출 #1227320

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1227320		vivkostov	Candies (JOI18_candies)	C++20	8 / 100	33 ms	26028 KiB

candies

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
void speed()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
}
struct cell
{
    long long int pos,st,type,to;
    bool operator<(const cell&a)const
    {
        return st<a.st;
    }
};
long long int n,a[200005];
long long int sum[200005],otg,dp[2005][2005],ot[2005];
map<int,int>m;
priority_queue<cell>q;
void prec()
{
    sum[1]=a[1];
    for(int i=2; i<=n; i++)
    {
        sum[i]=sum[i-2]+a[i];
    }
}
void er(int val)
{
    if(m[val]==0)m.erase(val);
}
void merg(int a,int b)
{
    m[b]=m[a];
    m.erase(a);
}
void fix(int a)
{
    m[a+1]=m[a]-1;
    m.erase(a);
}
void check(cell&w)
{
    auto up=m.upper_bound(w.pos);
    if(m[m[w.pos]-2])
    {
        merg(m[w.pos]-2,w.pos);
    }
    else er(m[w.pos]-2);
    if(up->second==w.pos+2)
    {
        merg(w.pos,up->first);
        w.pos=up->first;
    }
}
void add(cell w)
{
    if(w.pos==n||m[w.pos]==1)return;
    cell c;
    c.pos=w.pos;
    c.to=m[w.pos];
    c.type=2;
    c.st=sum[w.pos+1]-sum[w.pos]+sum[m[w.pos]-2];
    if(m[w.pos]>2)c.st-=sum[m[w.pos]-3];
    //cout<<c.pos<<" "<<c.to<<" "<<c.st<<" "<<c.type<<" "<<w.pos<<endl;
    q.push(c);
}
void resh()
{
    cell w;
    while(q.size())
    {
        w=q.top();
        q.pop();
        if(w.type==1)
        {
            auto up=m.upper_bound(w.pos);
            if(m[w.pos-1]||up->second==w.pos+1||(up->second<w.pos&up->first>w.pos))
            {
                er(w.pos-1);
                continue;
            }
            er(w.pos-1);
            otg+=w.st;
            m[w.pos]=w.pos;
            check(w);
        }
        if(w.type==2)
        {
            if(m[w.pos]!=w.to)
            {
                er(w.pos);
                continue;
            }
            otg+=w.st;
            fix(w.pos);
            w.pos++;
            check(w);
        }
        add(w);
        cout<<otg<<endl;
    }
    cout<<endl;
}
void make_dp()
{
    dp[1][1]=a[1];
    dp[2][1]=max(a[1],a[2]);
    for(int i=3;i<=n;i++)
    {
        for(int j=1;j<=(i+1)/2;j++)
        {
            dp[i][j]=max(dp[i-1][j],max(dp[i-2][j-1],dp[i-3][j-1])+a[i]);
            ot[j]=max(ot[j],dp[i][j]);
        }
    }
    for(int i=1;i<=(n+1)/2;i++)
    {
        cout<<ot[i]<<endl;
    }
}
void read()
{
    cell c;
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        cin>>a[i];
        c.pos=i;
        c.st=a[i];
        c.type=1;
        c.to=0;
        q.push(c);
    }
    cout<<endl;
    prec();
    make_dp();
    //resh();
}
int main()
{
    speed();
    read();
    return 0;
}

• Subtask 1: 8 / 8
• Subtask 2: 0 / 92