This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define db double
#define ii pair<int,int>
#define f first
#define s second
#define mp make_pair
#define mt make_tuple
#define pb push_back
#define all(v) v.begin(),v.end()
#define BIT(i) ((ll)1<<(i))
#define endl "\n"
#define debug(x) for (auto p: x) cout<<p<<' ';cout<<endl
#define forw(i,j,z) for(int i=(int)j;i<=(int)z;i++)
#define forw2(i,j,z,k) for(int i=(int)j;i<=(int)z;i+=k)
#define ford(i,j,z) for (int i=(int)j;i>=(int)z;i--)
#define ford2(i,j,z,k) for (int i=(int)j;i>=(int)z;i-=k)
#define sz(a) (int)a.size()
#define len(a) (int)a.length()
const ll inf=(ll)1<<60;
const int N=5e5;
const int Q=2e5;
int n;
vector<int> DS[N+1];
struct IT_tap{
IT_tap *l= nullptr;
IT_tap *r= nullptr;
vector<int> tap;
IT_tap():l(nullptr),r(nullptr){}
IT_tap(IT_tap *L, IT_tap *R):l(L),r(R){}
IT_tap(IT_tap *L, IT_tap *R,vector<int> &u):l(L),r(R),tap(u){}
};
IT_tap * tree_tap=new IT_tap();
struct IT{
IT *l= nullptr;
IT *r= nullptr;
int use=0;
IT():l(nullptr),r(nullptr){}
IT(IT *L, IT *R):l(L),r(R){}
};
IT *version[Q+10];
IT *build(int l = 1,int r = n, IT_tap *tree = tree_tap)
{
if (l==r){
sort(all(DS[l]));
tree->tap = DS[l];
return new IT(nullptr,nullptr);
}
int mid=(l+r)>>1;
tree->l=new IT_tap;
tree->r=new IT_tap;
IT * tmp= new IT(build(l,mid,tree->l),build(mid+1,r,tree->r));
tree->tap.assign(sz(tree->l->tap)+sz(tree->r->tap),0);
merge(all(tree->l->tap),all(tree->r->tap),begin(tree->tap));
return tmp;
}
void init(int _n,int A[],int B[])
{
n=_n;
forw(i,0,n-1){
DS[B[i]].push_back(A[i]);
}
version[0]=build();
}
IT *get(IT *tree,int k,int &left,int l = 1,int r = n, IT_tap *arr = tree_tap)
{
if (r < k||left == 0) return tree;
int i = upper_bound(all(arr->tap),k)-begin(arr->tap);
IT *newTree = new IT; *newTree = *tree;
int mid=(l+r)>>1;
if (i - newTree->use <= left)
{
left -= (i - newTree-> use);
newTree->use = i ;
return newTree;
}
else{
IT * __ = get(tree->l,k,left,l,mid,arr->l);
newTree->l = __;
if (left) {
__ = get(tree->r,k,left,mid+1,r,arr->r);
newTree->r = __;
}
return newTree;
}
}
int can(int m,int query[])
{
sort(query,query+m);
forw(i,0,m-1){
int k=query[i];
version[i+1]=get(version[i],query[i],k);
if (k>0) return 0;
}
return 1;
}
//signed main()
//{
// int A[5]={5,5,5,5,5};
// int B[5]={5,5,5,5,5};
// init(5,A,B);
//
// int query[1]={5};
// cout<<can(1,query)<<endl;
//
//}
Compilation message (stderr)
teams.cpp: In function 'IT* get(IT*, int, int&, int, int, IT_tap*)':
teams.cpp:69:41: warning: conversion from '__gnu_cxx::__normal_iterator<int*, std::vector<int> >::difference_type' {aka 'long int'} to 'int' may change value [-Wconversion]
69 | int i = upper_bound(all(arr->tap),k)-begin(arr->tap);
| ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |