Submission #1137375

#	Time	Username	Problem	Language	Result	Execution time	Memory
1137375		Zbyszek99	Exam (eJOI20_exam)	C++20	77 / 100	1096 ms	25320 KiB

exam

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long
#define ld long double
#define ull unsigned long long
#define ff first
#define ss second
#define pii pair<int,int>
#define pll pair<long long, long long>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define rep(i, b) for(int i = 0; i < (b); ++i)
#define rep2(i,a,b) for(int i = a; i <= (b); ++i)
#define rep3(i,a,b,c) for(int i = a; i <= (b); i+=c)
#define count_bits(x) __builtin_popcountll((x))
#define all(x) (x).begin(),(x).end()
#define siz(x) (int)(x).size()
#define forall(it,x) for(auto& it:(x))
using namespace __gnu_pbds;
using namespace std;
typedef tree<int, null_type, less<int>, rb_tree_tag,tree_order_statistics_node_update> ordered_set;
//mt19937 mt;void random_start(){mt.seed(chrono::time_point_cast<chrono::milliseconds>(chrono::high_resolution_clock::now()).time_since_epoch().count());}
//ll los(ll a, ll b) {return a + (mt() % (b-a+1));}
const int INF = 1e9+50;
const ll INF_L = 1e18+40;
const ll MOD = 1e9+7;

struct seg
{
    int l,r,x;
    bool type;
    bool operator<(const seg& s)
    {
        if(r != s.r) return r < s.r;
        return l < s.l;
    }
};

const int tree_siz = 1024*256-1;
int drzewo_max[tree_siz];

int get_max(int akt, int p1, int p2, int s1, int s2)
{
    if(p2 < s1 || p1 > s2) return -1;
    if(p1 >= s1 && p2 <= s2) return drzewo_max[akt-1];
    return max(get_max(akt*2,p1,(p1+p2)/2,s1,s2),get_max(akt*2+1,(p1+p2)/2+1,p2,s1,s2)); 
}

int A[100001];
int B[100001];
set<int> A_poz[200001];
vector<seg> segs;
int ans[100001];

int main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    //random_start();
    int n;
    cin >> n;
    map<int,int> m;
    rep(i,n) cin >> A[i];
    rep(i,n) cin >> B[i];
    set<int> sB;
    rep(i,n) sB.insert(B[i]);
    if(siz(sB) == 1)
    {
        int ok_count = 0;
        rep(i,n)
        {
            if(A[i] == B[0])
            {
                ok_count++;
                for(int j = i-1; j >= 0; j--)
                {
                    if(A[j] >= B[0] || A[j] == -1)
                    {
                        break;
                    }
                    ok_count++;
                    A[j] = -1;
                }
                for(int j = i+1; j < n; j++)
                {
                    if(A[j] >= B[0] || A[j] == -1)
                    {
                        break;
                    }
                    ok_count++;
                    A[j] = -1;
                }
            }
        }
        cout << ok_count << "\n";
        return 0;
    }
    rep(i,n) m[A[i]]++;
    rep(i,n) m[B[i]]++;
    int cur = 0;
    forall(it,m)
    {
        m[it.ff] = cur++;
    }
    rep(i,n) A[i] = m[A[i]];
    rep(i,n) B[i] = m[B[i]];
    rep(i,n) A_poz[A[i]].insert(i);
    rep(i,n)
    {
        drzewo_max[i+tree_siz/2] = A[i];
    }
    for(int i = tree_siz/2-1; i >= 0; i--)
    {
        drzewo_max[i] = max(drzewo_max[i*2+1],drzewo_max[i*2+2]);
    }
    rep(i,n)
    {
       // cout << A[i] << " " << B[i] << " ab\n";
        auto it_pop = A_poz[B[i]].upper_bound(i);
        if(it_pop != A_poz[B[i]].begin())
        {
            it_pop--;
            if(get_max(1,0,tree_siz/2,*it_pop,i) == B[i])
            {
                segs.pb({*it_pop,i,B[i],0});
            }
        }
        auto it_next = A_poz[B[i]].lower_bound(i);
        if(B[i] != A[i] && it_next != A_poz[B[i]].end())
        {
            if(get_max(1,0,tree_siz/2,i,*it_next) == B[i])
            {
                segs.pb({i,*it_next,B[i],1});
            }
        }
    }
    sort(all(segs));
 //   forall(it,segs)
  //  {
  //      cout << it.l << " " << it.r << " " << it.type << " it\n";
 //   }
    int final_ans = 0;
    rep(i,siz(segs))
    {
        ans[i] = 1;
        rep(j,i)
        {
            if(segs[i].type != segs[j].type)
            {
                if(segs[i].l > segs[j].r || (segs[i].type == 0 && segs[i].l == segs[j].r))
                {
                    ans[i] = max(ans[i],ans[j]+1);
                //    cout << j << " j1\n";
                }
            }
            else
            {
                if(segs[i].x == segs[j].x)
                {
                    ans[i] = max(ans[i],ans[j]+1);
                  //  cout << j << " j2\n";
                }
                else
                {
                    if(segs[i].l >= segs[j].l)
                    {
                        ans[i] = max(ans[i],ans[j]+1);
                      //  cout << j << " j3\n";
                    }
                }
            }
        }
      //  cout << ans[i] << " ans\n";
        final_ans = max(final_ans,ans[i]);
    }
    cout << final_ans << "\n";
}

• Subtask 1: 14 / 14
• Subtask 2: 12 / 12
• Subtask 3: 13 / 13
• Subtask 4: 0 / 23
• Subtask 5: 16 / 16
• Subtask 6: 22 / 22