oj.uz

Submission #1043788

# Submission time Handle Problem Language Result Execution time Memory
1043788 2024-08-04T17:34:13 Z Dan4Life Distributing Candies (IOI21_candies) C++17
0 / 100
 54 ms 12616 KB 
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include "candies.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)
using vi = vector<int>;
using ll = long long;
const int mxN = (int)2e5+10;
const int INF = (int)1e9+1;
const ll LINF = (ll)1e18;
ll mxsuf[mxN], mnsuf[mxN], pref[mxN];

vi distribute_candies(vi c, vi l, vi r, vi v){
    int n = sz(c), q = sz(l); vector<ll> ans(n,0);
    l.insert(begin(l),0), l.insert(begin(l),0);
    r.insert(begin(r),n-1); r.insert(begin(r),n-1);
    v.insert(begin(v),-INF); v.insert(begin(v),INF); q+=2;
    if(*max_element(all(l))==0 and *min_element(all(r))==n-1){
		mxsuf[q]=-LINF; mnsuf[q]=LINF; pref[0]=0;
		for(int i = 0; i < q; i++) pref[i]=(i?pref[i-1]:0)+v[i];
		for(int i = q-1; i >= 0; i--) mxsuf[i]=max(mxsuf[i+1],pref[i]);
		for(int i = q-1; i >= 0; i--) mnsuf[i]=min(mnsuf[i+1],pref[i]);
		for(int i = 0; i < n; i++){
			int k = 0;
			for(int j = 0; j < q; j++){
				ans[i]+=v[j];
				if(ans[i]<=0) ans[i]=0,k=-(j+1);
				if(ans[i]>=c[i]) ans[i]=c[i],k=(j+1);
			}
			ans[i] = 0;
			if(mxsuf[0]-mnsuf[0] < c[i]){
				ans[i]+=pref[q]; continue;
			}
			int L = 0, R = q-1;
			while(L<R){
				int mid = (L+R)/2;
				if(mxsuf[mid]-mnsuf[mid] >= c[i]) L=mid+1;
				else R=mid;
			}
			if(v[L]>0) ans[i]=c[i];
			ans[i]+=pref[q-1]-pref[L];
		}
	}
	vi Ans; Ans.clear();
	for(auto u : ans) Ans.pb(u);
    return Ans;
}

Compilation message

candies.cpp: In function 'vi distribute_candies(vi, vi, vi, vi)':
candies.cpp:27:8: warning: variable 'k' set but not used [-Wunused-but-set-variable]
   27 |    int k = 0;
      |        ^

Subtask #1
0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4444 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 54 ms 12616 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 27.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 29.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 4444 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4444 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -