oj.uz

Submission #831048

# Submission time Handle Problem Language Result Execution time Memory
831048 2023-08-19T16:00:25 Z NothingXD Distributing Candies (IOI21_candies) C++17
8 / 100
 1627 ms 11716 KB 
#include "candies.h"
#include <bits/stdc++.h>

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx")

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

void debug_out(){cerr << endl;}
template<typename Head, typename... Tail>
void debug_out(Head H, Tail... T){
	cerr << H << ' ';
	debug_out(T...);
}

#define debug(...) cerr << "(" << #__VA_ARGS__ << "): ", debug_out(__VA_ARGS__)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define MP(x, y) make_pair(x, y)

const int maxn = 2e5 + 10;

int n, q, a[maxn], c[maxn], l[maxn], r[maxn], v[maxn];

void add1(int l, int r, int x){
//	debug(1, l, r, x);
	for (int i = l; i <= r; i++){
		a[i] = (a[i]+x > c[i]? c[i]: a[i]+x);
	}
}

void add2(int l, int r, int x){
//	debug(2, l, r, x);
	for (int i = l; i <= r; i++){
		a[i] = (a[i]+x < 0? 0: a[i]+x);
	}
}

void add3(int l, int r, int x, int y){
//	debug(3, l, r, x, y);
	for (int i = l; i <= r; i++){
		a[i] = (a[i]+x > c[i]? c[i]: a[i]+x);
//		debug(i, a[i]);
		a[i] = (a[i]+y < 0? 0: a[i]+y);
//		debug(i, a[i]);
	}
}

void add4(int l, int r, int x, int y){
	//debug(4, l, r, x, y);
	for (int i = l; i <= r; i++){
		a[i] = (a[i]+x < 0? 0: a[i]+x);
		a[i] = (a[i]+y < c[i]? c[i]: a[i]+y);
	}
}

vector<int> distribute_candies(vector<int> C, vector<int> L, vector<int> R, vector<int> V) {
	n = C.size();
	q = L.size();
	for (int i = 0; i < n; i++){
		c[i] = C[i];
	}
	for (int i = 0; i < q; i++){
		l[i] = L[i];
		r[i] = R[i];
		v[i] = V[i];
	}
	for (int i = 0; i < q; i += 2){
		if (i + 1 == q){
			if (v[i] > 0) add1(l[i], r[i], v[i]);
			if (v[i] < 0) add2(l[i], r[i], v[i]);
		}
		if (v[i] == 0 && v[i+1] == 0) continue;
		if (v[i] >= 0 && v[i+1] >= 0){
			add1(l[i], min(r[i], l[i+1]-1), v[i]);
			add1(l[i+1], min(r[i+1], l[i]-1), v[i+1]);
			add1(max(l[i], l[i+1]), min(r[i], r[i+1]), v[i]+v[i+1]);
			add1(max(l[i], r[i+1]+1), r[i], v[i]);
			add1(max(l[i+1], r[i]+1), r[i+1], v[i+1]);
		}
		else if (v[i] <= 0 && v[i+1] <= 0){
			add2(l[i], min(r[i], l[i+1]-1), v[i]);
			add2(l[i+1], min(r[i+1], l[i]-1), v[i+1]);
			add2(max(l[i], l[i+1]), min(r[i], r[i+1]), v[i]+v[i+1]);
			add2(max(l[i], r[i+1]+1), r[i], v[i]);
			add2(max(l[i+1], r[i]+1), r[i+1], v[i+1]);
		}
		else if (v[i] >= 0 && v[i+1] <= 0){
			add1(l[i], min(r[i], l[i+1]-1), v[i]);
			add2(l[i+1], min(r[i+1], l[i]-1), v[i+1]);
			add3(max(l[i], l[i+1]), min(r[i], r[i+1]), v[i], v[i+1]);
			add1(max(l[i], r[i+1]+1), r[i], v[i]);
			add2(max(l[i+1], r[i]+1), r[i+1], v[i+1]);
		}
		else{
			add2(l[i], min(r[i], l[i+1]-1), v[i]);
			add1(l[i+1], min(r[i+1], l[i]-1), v[i+1]);
			add4(max(l[i], l[i+1]), min(r[i], r[i+1]), v[i], v[i+1]);
			add2(max(l[i], r[i+1]+1), r[i], v[i]);
			add1(max(l[i+1], r[i]+1), r[i+1], v[i+1]);
		}
	}
	vector<int> ans;
	for (int i = 0; i < n; i++){
		ans.push_back(a[i]);
	}
	return ans;
}

Subtask #1
0 / 3.0

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

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 1627 ms 11632 KB Output is correct
2 Correct 1608 ms 11716 KB Output is correct
3 Correct 1553 ms 11664 KB Output is correct

Subtask #3
0 / 27.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 340 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 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 33.0

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