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>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template<typename T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
typedef long long int ll;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
#define fastio ios_base::sync_with_stdio(false); cin.tie(NULL)
#define pb push_back
#define endl '\n'
#define sz(a) (int)a.size()
#define setbits(x) __builtin_popcountll(x)
#define ff first
#define ss second
#define conts continue
#define ceil2(x,y) ((x+y-1)/(y))
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define yes cout << "Yes" << endl
#define no cout << "No" << endl
#define rep(i,n) for(int i = 0; i < n; ++i)
#define rep1(i,n) for(int i = 1; i <= n; ++i)
#define rev(i,s,e) for(int i = s; i >= e; --i)
#define trav(i,a) for(auto &i : a)
template<typename T>
void amin(T &a, T b) {
a = min(a,b);
}
template<typename T>
void amax(T &a, T b) {
a = max(a,b);
}
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) 42
#endif
/*
refs:
edi
*/
const int MOD = 1e9 + 7;
const int N = 1e5 + 5;
const int inf1 = int(1e9) + 5;
const ll inf2 = ll(1e18) + 5;
template<typename T>
struct fenwick {
int n;
vector<T> tr;
int LOG = 0;
fenwick() {
}
fenwick(int n_) {
n = n_;
tr = vector<T>(n + 1);
while((1<<LOG) <= n) LOG++;
}
void reset(){
fill(all(tr),0);
}
int lsb(int x) {
return x & -x;
}
void pupd(int i, T v) {
for(; i <= n; i += lsb(i)){
tr[i] += v;
}
}
T sum(int i) {
T res = 0;
for(; i; i ^= lsb(i)){
res += tr[i];
}
return res;
}
T query(int l, int r) {
if (l > r) return 0;
T res = sum(r) - sum(l - 1);
return res;
}
int lower_bound(T s){
// first pos with sum >= s
if(sum(n) < s) return n+1;
int i = 0;
rev(bit,LOG-1,0){
int j = i+(1<<bit);
if(j > n) conts;
if(tr[j] < s){
s -= tr[j];
i = j;
}
}
return i+1;
}
int upper_bound(T s){
return lower_bound(s+1);
}
};
void solve(int test_case)
{
ll n,k; cin >> n >> k;
vector<ll> a(n+5), b(n+5);
rep1(i,n) cin >> a[i];
rep1(i,n) cin >> b[i];
vector<ll> p(n+5);
rep1(i,n) p[i] = p[i-1]+a[i];
multiset<ll> ms1,ms2;
ll curr_sum = 0;
ll lx = 1, rx = 0;
auto transfer = [&](){
while(!ms2.empty() and sz(ms1) < k){
curr_sum += *ms2.rbegin();
ms1.insert(*ms2.rbegin());
ms2.erase(--ms2.end());
}
while(sz(ms1) > k){
ll x = *ms1.begin();
curr_sum -= x;
ms1.erase(ms1.begin());
ms2.insert(x);
}
if(ms2.empty()) return;
while(true){
ll mn1 = *ms1.begin(), mx2 = *ms2.rbegin();
if(mn1 >= mx2) break;
ms1.erase(ms1.find(mn1));
ms2.erase(ms2.find(mx2));
ms1.insert(mx2);
ms2.insert(mn1);
curr_sum += mx2-mn1;
}
};
auto ins = [&](ll i){
ms1.insert(b[i]);
curr_sum += b[i];
transfer();
};
auto del = [&](ll i){
if(ms1.find(b[i]) != ms1.end()){
ms1.erase(ms1.find(b[i]));
curr_sum -= b[i];
}
else{
ms2.erase(ms2.find(b[i]));
}
transfer();
};
auto f = [&](ll l, ll r){
if(r-l+1 < k) return -inf2;
// expand
while(rx < r){
rx++;
ins(rx);
}
while(lx > l){
lx--;
ins(lx);
}
// contract
while(rx > r){
del(rx);
rx--;
}
while(lx < l){
del(lx);
lx++;
}
ll sum = -(p[r]-p[l-1]);
sum += curr_sum;
return sum;
};
ll ans = -inf2;
vector<pll> segs;
auto upd = [&](ll l, ll r, ll x){
if(x < ans) return;
if(x > ans){
segs.clear();
}
ans = x;
segs.pb({l,r});
};
auto go = [&](ll l, ll r, ll optl, ll optr, auto &&go) -> void{
if(l > r) return;
ll mid = (l+r) >> 1;
ll best = -inf2, optm = -1;
for(int i = optl; i <= optr; ++i){
ll cost = f(mid,i);
if(cost >= best){
best = cost;
optm = i;
}
upd(mid,i,cost);
}
go(l,mid-1,optl,optm,go);
go(mid+1,r,optm,optr,go);
};
go(1,n-k+1,1,n,go);
cout << ans << endl;
vector<ll> c;
rep1(i,n) c.pb(b[i]);
c.pb(-1);
sort(all(c));
c.resize(unique(all(c))-c.begin());
vector<ll> cb(n+5);
rep1(i,n) cb[i] = lower_bound(all(c),b[i])-c.begin();
vector<ll> pos[n+5];
rep1(i,n) pos[cb[i]].pb(i);
vector<array<ll,3>> here[n+5];
ll siz = sz(segs);
rep(i,siz){
auto [l,r] = segs[i];
here[(1+n)>>1].pb({1,n,i});
}
vector<ll> kth(siz,-1);
fenwick<ll> fenw(n+5);
while(true){
vector<array<ll,3>> nxt;
fenw.reset();
rev(mid,n,1){
trav(i,pos[mid]){
fenw.pupd(i,1);
}
for(auto [lo,hi,i] : here[mid]){
auto [l,r] = segs[i];
ll cnt = fenw.query(l,r);
if(cnt >= k){
kth[i] = mid;
nxt.pb({mid+1,hi,i});
}
else{
nxt.pb({lo,mid-1,i});
}
}
}
rep1(i,n) here[i].clear();
bool ok = false;
for(auto [lo,hi,i] : nxt){
if(lo > hi) conts;
ok = true;
here[(lo+hi)>>1].pb({lo,hi,i});
}
if(!ok) break;
}
vector<ll> enter[n+5], leave[n+5];
rep(i,siz){
auto [l,r] = segs[i];
enter[l].pb(kth[i]);
leave[r+1].pb(kth[i]);
}
multiset<ll> ms;
rep1(i,n){
trav(x,enter[i]){
ms.insert(x);
}
trav(x,leave[i]){
ms.erase(ms.find(x));
}
if(!ms.empty() and cb[i] >= *ms.begin()){
cout << 1;
}
else{
cout << 0;
}
}
cout << endl;
}
int main()
{
fastio;
int t = 1;
// cin >> t;
rep1(i, t) {
solve(i);
}
return 0;
}
Compilation message (stderr)
trade.cpp: In function 'void solve(int)':
trade.cpp:261:14: warning: structured binding declaration set but not used [-Wunused-but-set-variable]
261 | auto [l,r] = segs[i];
| ^~~~~
# | 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... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |