oj.uz

Submission #1025263

# Submission time Handle Problem Language Result Execution time Memory
1025263 2024-07-16T18:46:11 Z NintsiChkhaidze Road Construction (JOI21_road_construction) C++17
65 / 100
 10000 ms 145584 KB 
#include <bits/stdc++.h>
#define pb push_back
#define s second
#define f first
#define pb push_back
#define pii pair <int,int>
#define ll long long
#define left (h<<1),l,((l + r) >> 1)
#define right ((h<<1)|1),((l + r)>>1) + 1,r
#pragma GCC optimize("Ofast,unroll-loops")
// #define int ll
#define pli pair <ll,int>
#define pll pair <ll,ll>
using namespace std;
 
const int N = 3e5 + 5;
const ll inf = 1e18;
pli t[2][3*N];
int x[N],y[N],yy[N],n,k;
multiset <ll> st[2][3*N];
pii arr[N];
vector <ll> answ;
 
void upd(bool i,int h,int l,int r,int idx,ll oldval,ll newval,bool z){
  if (l == r){
    if (!z) st[i][l].erase(st[i][l].find(oldval));
    st[i][l].insert(newval);
    t[i][h] = {*--st[i][l].end(),l};
    return;
  }
 
  if (idx > (l + r)/2) upd(i,right,idx, oldval, newval,z);
  else upd(i,left,idx, oldval, newval,z);
  t[i][h] = max(t[i][h*2],t[i][h*2 + 1]);
}
 
void build(int h,int l,int r){
  t[1][h] = t[0][h] = {-inf,0};
  if (l == r) {
    st[1][l].clear();
    st[0][l].clear();
    return;
  }
  build(left);
  build(right);
}
 
pli get(bool i,int h,int l,int r,int L,int R){
  if (r < L || R < l) return {-inf,0};
  if (L <= l && r <= R) return t[i][h];
  return max(get(i,left,L,R),get(i,right,L,R));
}
 
int count(ll w,bool z){
 
  build(1,1,n);
  int tot = 0;
  for (int i = 1; i <= n; i++){
    if (tot > k) return tot;
    vector <pair <pii,pll> > roll;
 
    int id = arr[i].s;
    int v1 = x[id] + y[id];
    while (tot <= k){
      pli res = get(0,1,1,n,1,yy[id]);
      ll w1 = v1 - res.f;
      if (w1 > w) break;
      if (z) answ.pb(w1);
 
      tot += 1;
      roll.pb({{0,res.s},{res.f,-inf}});
      upd(0,1,1,n,res.s,res.f, -inf,0);
    }
 
    int v2 = x[id] - y[id];
    while (tot <= k){
      pli res = get(1,1,1,n,yy[id] + 1,n);
      ll w1 = v2 - res.f;
      if (w1 > w) break;
      if (z) answ.pb(w1);
 
      tot += 1;
      roll.pb({{1,res.s},{res.f,-inf}});
      upd(1,1,1,n,res.s,res.f, -inf,0);
    }
 
    for (int j = roll.size() - 1; j >= 0; j--){
      upd(roll[j].f.f,1,1,n,roll[j].f.s,roll[j].s.s,roll[j].s.f,0);
    }
 
    upd(0,1,1,n,yy[id],-inf,v1,1);
    upd(1,1,1,n,yy[id],-inf,v2,1);
  }
 
  return tot;
}
 
signed main() {
  ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
	
  cin>>n>>k;
  
  vector <int> vec;
  for (int i = 1; i <= n; i++){
    cin >> x[i] >> y[i];
    vec.pb(y[i]);
  }
  sort(vec.begin(),vec.end());
  map <int,int> mp;
  int val = 0;
  for (int i=0;i<vec.size();i++){
    if (!i || vec[i] != vec[i - 1]) ++val;
    mp[vec[i]] = val;
  }
 
  for (int i = 1; i<= n; i++){
    yy[i] = mp[y[i]];
    arr[i] = {x[i],i};
  }
  sort(arr+1,arr+n+1);
 
  ll l = 0,r = 4e9,res = -1;
  while (l <= r){
    ll mid = (l + r) >> 1;
    //umciresi w iseti rom w ze naklebi an toli wonis wiboebis raodenoba >= k ze
  
    if (count(mid,0) >= k) {
      res = mid;
      r = mid - 1;
    }else{
      l = mid + 1;
    }
  }
 
  int cnt1 = count(res - 1,1);
  sort(answ.begin(),answ.end());
  for (int i = 1; i <= k - cnt1; i++)
    answ.pb(res);
    
  for (ll x: answ)
    cout<<x<<endl;
}

Compilation message

road_construction.cpp: In function 'int main()':
road_construction.cpp:111:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  111 |   for (int i=0;i<vec.size();i++){
      |                ~^~~~~~~~~~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 2407 ms 89852 KB Output is correct
2 Correct 2437 ms 89888 KB Output is correct
3 Correct 2458 ms 90064 KB Output is correct
4 Correct 2493 ms 90064 KB Output is correct
5 Correct 2571 ms 88908 KB Output is correct
6 Correct 69 ms 85328 KB Output is correct

Subtask #2
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 8083 ms 134520 KB Output is correct
2 Correct 8115 ms 134008 KB Output is correct
3 Correct 2907 ms 89816 KB Output is correct
4 Correct 9159 ms 135788 KB Output is correct
5 Correct 7980 ms 134036 KB Output is correct
6 Correct 7242 ms 133936 KB Output is correct
7 Correct 6179 ms 136588 KB Output is correct

Subtask #3
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 5167 ms 145580 KB Output is correct
2 Correct 5036 ms 145580 KB Output is correct
3 Correct 17 ms 90716 KB Output is correct
4 Correct 1860 ms 133708 KB Output is correct
5 Correct 3554 ms 133836 KB Output is correct

Subtask #4
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 5167 ms 145580 KB Output is correct
2 Correct 5036 ms 145580 KB Output is correct
3 Correct 17 ms 90716 KB Output is correct
4 Correct 1860 ms 133708 KB Output is correct
5 Correct 3554 ms 133836 KB Output is correct
6 Correct 5465 ms 145356 KB Output is correct
7 Correct 5412 ms 145584 KB Output is correct
8 Correct 14 ms 90716 KB Output is correct
9 Correct 15 ms 90716 KB Output is correct
10 Correct 5297 ms 144076 KB Output is correct
11 Correct 1643 ms 133844 KB Output is correct
12 Correct 3620 ms 133712 KB Output is correct

Subtask #5
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 2407 ms 89852 KB Output is correct
2 Correct 2437 ms 89888 KB Output is correct
3 Correct 2458 ms 90064 KB Output is correct
4 Correct 2493 ms 90064 KB Output is correct
5 Correct 2571 ms 88908 KB Output is correct
6 Correct 69 ms 85328 KB Output is correct
7 Correct 6438 ms 119748 KB Output is correct
8 Correct 6309 ms 119724 KB Output is correct
9 Correct 2339 ms 95616 KB Output is correct
10 Correct 4913 ms 117936 KB Output is correct
11 Correct 4950 ms 117592 KB Output is correct
12 Correct 4984 ms 114880 KB Output is correct
13 Correct 6053 ms 113544 KB Output is correct

Subtask #6
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 2407 ms 89852 KB Output is correct
2 Correct 2437 ms 89888 KB Output is correct
3 Correct 2458 ms 90064 KB Output is correct
4 Correct 2493 ms 90064 KB Output is correct
5 Correct 2571 ms 88908 KB Output is correct
6 Correct 69 ms 85328 KB Output is correct
7 Correct 8083 ms 134520 KB Output is correct
8 Correct 8115 ms 134008 KB Output is correct
9 Correct 2907 ms 89816 KB Output is correct
10 Correct 9159 ms 135788 KB Output is correct
11 Correct 7980 ms 134036 KB Output is correct
12 Correct 7242 ms 133936 KB Output is correct
13 Correct 6179 ms 136588 KB Output is correct
14 Correct 5167 ms 145580 KB Output is correct
15 Correct 5036 ms 145580 KB Output is correct
16 Correct 17 ms 90716 KB Output is correct
17 Correct 1860 ms 133708 KB Output is correct
18 Correct 3554 ms 133836 KB Output is correct
19 Correct 5465 ms 145356 KB Output is correct
20 Correct 5412 ms 145584 KB Output is correct
21 Correct 14 ms 90716 KB Output is correct
22 Correct 15 ms 90716 KB Output is correct
23 Correct 5297 ms 144076 KB Output is correct
24 Correct 1643 ms 133844 KB Output is correct
25 Correct 3620 ms 133712 KB Output is correct
26 Correct 6438 ms 119748 KB Output is correct
27 Correct 6309 ms 119724 KB Output is correct
28 Correct 2339 ms 95616 KB Output is correct
29 Correct 4913 ms 117936 KB Output is correct
30 Correct 4950 ms 117592 KB Output is correct
31 Correct 4984 ms 114880 KB Output is correct
32 Correct 6053 ms 113544 KB Output is correct
33 Execution timed out 10074 ms 145356 KB Time limit exceeded
34 Halted 0 ms 0 KB -