Submission #613622

#	Time	Username	Problem	Language	Result	Execution time	Memory
613622		andrei_boaca	Cake (CEOI14_cake)	C++14	46.67 / 100	2096 ms	13392 KiB

cake

#include <bits/stdc++.h>

using namespace std;
typedef pair<int,int> pii;
int n,q,a,d[300005];
set<pii> s;
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>a;
    for(int i=1;i<=n;i++)
    {
        cin>>d[i];
        s.insert({d[i],i});
    }
    d[0]=d[n+1]=1e9;
    cin>>q;
    while(q--)
    {
        char c;
        cin>>c;
        if(c=='F')
        {
            int poz;
            cin>>poz;
            if(poz==a)
            {
                cout<<0<<'\n';
                continue;
            }
            int st=a-1;
            int dr=a+1;
            int cnt=1;
            while(true)
            {
                cnt++;
                if(d[st]<d[dr])
                {
                    if(st==poz)
                    {
                        cout<<cnt-1<<'\n';
                        break;
                    }
                    st--;
                }
                else
                {
                    if(dr==poz)
                    {
                        cout<<cnt-1<<'\n';
                        break;
                    }
                    dr++;
                }
            }
        }
        else
        {
            int poz,val;
            cin>>poz>>val;
            int old=d[poz];
            vector<int> mypoz;
            auto it=prev(s.end());
            int cnt=0;
            int myval=0;
            while(cnt+1<val)
            {
                cnt++;
                int p=(*it).second;
                myval=(*it).first;
                mypoz.push_back(p);
                it--;
            }
            if(myval==0)
                myval=(*it).first+1;
            s.erase({d[poz],poz});
            d[poz]=myval;
            s.insert({d[poz],poz});
            for(auto i:mypoz)
            {
                s.erase({d[i],i});
                d[i]++;
                s.insert({d[i],i});
            }
        }
    }
    return 0;
}

Compilation message (stderr)

cake.cpp: In function 'int main()':
cake.cpp:62:17: warning: unused variable 'old' [-Wunused-variable]
   62 |             int old=d[poz];
      |                 ^~~

• Subtask 1: 15 / 15
• Subtask 2: 15 / 15
• Subtask 3: 0 / 20
• Subtask 4: 16.666667 / 50