| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 769610 | YassirSalama | Distributing Candies (IOI21_candies) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "candies.h"
#include<bits/stdc++.h>
#include <vector>
using namespace std;
const int MAXN=2e5;
long long tree[4*MAXN+1];
long long lazy[4*MAXN+1];
vector<int> cc;
vector<int> ans;
void push(int node,int l,int r){
if(lazy[node]!=0){
tree[node]+=lazy[node];
if(l!=r) lazy[node<<1]+=lazy[node],
lazy[node<<1|1]+=lazy[node];
if(l==r) tree[node]=min(tree[node],(long long)cc[l]);
lazy[node]=0;
}
}
void update(int node,int l,int r,int ql,int qr,int v){
push(node,l,r);
if(l==r){
tree[node]=min((long long)cc[l],tree[node]+v);
return;
}
if(ql<=l&&r<=qr){
lazy[node]+=v;
push(node,l,r);
return;
}
int mid=(l+r)/2;
if(ql<=mid) update(node<<1,l,mid,ql,qr,v);
if(qr>mid) update(node<<1|1,mid+1,r,ql,qr,v);
push(node<<1,l,mid);
push(node<<1|1,mid+1,r);
tree[node]=min(tree[node<<1],tree[node<<1|1]);
}
void build(int node,int l,int r){
push(node,l,r);
if(l==r){
push(node,l,r);
ans[l]=(min(tree[node],(long long)cc[l]));
return;
}
int mid=(l+r)/2;
build(node<<1,l,mid);
build(node<<1|1,mid+1,r);
}
vector<int> distribute_candies(vector<int> c, vector<int> l,
vector<int> r, vector<int> v) {
// memset(lazy,0,sizeof lazy);
// memset(tree,0,sizeof tree);
for(int i=0;i<4*MAXN;i++) tree[i]=0,lazy[i]=0;
int n = c.size();
int q = v.size();
ans.resize(n,0);
cc=c;
if(n<=2000&&q<=2000){
for(int j=0;j<q;j++){
int lrr=l[j],rr=r[j],cc=v[j];
for(int i=lrr;i<=rr;i++){
s[i]=max(0LL,min((long long)c[i],s[i]+cc));
}
}
for(int i=0;i<n;i++) ans[i]=(int)s[i];
}else{
for(int j=0;j<q;j++){
int lrr=l[j],rr=r[j],cc=v[j];
update(1,0,n-1,lrr,rr,cc);
}
build(1,0,n-1);
}
return ans;
}