| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 448006 | aris12345678 | 사탕 분배 (IOI21_candies) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "candies.h"
using namespace std;
const int mxN = 200005;
bool sub2(vector<int> v) {
int q = int(v.size());
for(int i = 0; i < q; i++) {
if(v[i] < 0)
return false;
}
return true;
}
vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v) {
int n = int(c.size()), q = int(l.size());
vector<int> ans(n, 0);
vector<int> pref(n, 0);
if(sub2(v)) {
for(int i = 0; i < q; i++) {
pref[l[i]] += v[i]
if(r[i] < n)
pref[r[i]] -= v[i];
}
for(int i = 0; i < n; i++) {
if(i > 0)
pref[i] += pref[i-1];
ans[i] = min(c[i], pref[i]);
}
} else {
for(int i = 0; i < q; i++) {
for(int j = l[i]; j <= r[i]; j++) {
if(v[i] > 0)
ans[j] = min(c[j], ans[j]+v[i]);
else
ans[j] = max(0, ans[j]+v[i]);
}
}
}
return ans;
}
