#include "candies.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v) {
int n = c.size(), q = v.size();
vector<int64_t>dif(n+1,0);
bool pos=true, same = true;
for(int i = 0 ; i < q ; i++) {
if(v[i]<0) {
pos=false;
}
if(l[i] != r[i]) same=false;
}
vector<int>ans(n,0);
if(pos) {
for(int i = 0 ; i < q ; i++) {
dif[l[i]]+=v[i],dif[r[i]+1]-=v[i];
}
int64_t curr=0;
for(int i = 0 ; i < n ; i++) {
curr+=dif[i];
ans[i]=min((int64_t)c[i],curr);
}
} else if(same) {
vector<pair<int,int>>cap(n);
for(int i = 0 ; i < n ; i++) {
cap[i].first=c[i],cap[i].second=i;
}
sort(cap.begin(),cap.end());
int current_height=0,position=-1;
for(auto z : v) {
if(z>0) {
current_height+=z;
current_height=min(current_height,cap[n-1].first);
int left=0,right=n-1;
while(left <= right) {
int mid = (left+right)/2;
if(cap[mid].first <= current_height) position=mid,left=mid+1;
else right=mid-1;
}
} else {
current_height+=z;
current_height = max(current_height,0);
int left=0,right=n-1;
while(left <= right) {
int mid = (left+right)/2;
if(cap[mid].first <= current_height) position=mid,left=mid+1;
else right=mid-1;
}
}
}
for(int i = 0 ; i <= position ; i++) {
ans[cap[i].second] = cap[i].first;
}
for(int i = position+1 ; i < n ; i++) {
ans[cap[i].second] = cap[position].first;
}
} else {
for(int i = 0 ; i < q ; i++) {
for(int j = l[i] ; j <= r[i] ; j++) ans[j]=(v[i]>0?min(c[j],ans[j]+v[i]):max(0,ans[j]+v[i]));
}
}
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
2 ms |
332 KB |
Output is correct |
4 |
Correct |
2 ms |
304 KB |
Output is correct |
5 |
Correct |
5 ms |
332 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
139 ms |
8924 KB |
Output is correct |
2 |
Correct |
172 ms |
8820 KB |
Output is correct |
3 |
Correct |
134 ms |
8792 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
211 ms |
4948 KB |
Output is correct |
3 |
Correct |
210 ms |
4156 KB |
Output is correct |
4 |
Execution timed out |
5038 ms |
8800 KB |
Time limit exceeded |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
538 ms |
5036 KB |
Output is correct |
4 |
Correct |
529 ms |
4252 KB |
Output is correct |
5 |
Execution timed out |
5031 ms |
8796 KB |
Time limit exceeded |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
2 ms |
332 KB |
Output is correct |
4 |
Correct |
2 ms |
304 KB |
Output is correct |
5 |
Correct |
5 ms |
332 KB |
Output is correct |
6 |
Correct |
139 ms |
8924 KB |
Output is correct |
7 |
Correct |
172 ms |
8820 KB |
Output is correct |
8 |
Correct |
134 ms |
8792 KB |
Output is correct |
9 |
Correct |
1 ms |
204 KB |
Output is correct |
10 |
Correct |
211 ms |
4948 KB |
Output is correct |
11 |
Correct |
210 ms |
4156 KB |
Output is correct |
12 |
Execution timed out |
5038 ms |
8800 KB |
Time limit exceeded |
13 |
Halted |
0 ms |
0 KB |
- |