제출 #1043680

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1043680		42kangaroo	사탕 분배 (IOI21_candies)	C++17	0 / 100	5084 ms	37972 KiB

candies

#include "candies.h"
#include "bits/stdc++.h"

using namespace std;

struct Val {
	long long mi, miPo, ma, maPo;
};

Val operator+(const Val& l, const Val& r) {
	Val re{min(l.mi, r.mi), 0, max(l.ma, r.ma), 0};
	if (l.mi < r.mi) re.miPo = l.miPo;
	else re.miPo = r.miPo;
	if (l.ma >= r.ma) re.maPo = l.maPo;
	else re.maPo = r.maPo;
	return re;
}

struct SegTe {
	vector<int> la;
	vector<Val> a;

	void push(int i) {
		if (la[i] != 0) {
			a[i].mi += la[i];
			a[i].ma += la[i];
			if (2*i + 2 < a.size()) {
				la[2*i + 1] += la[i];
				la[2*i + 2] += la[i];
			}
			la[i] = 0;
		}
	}

	Val build(int l, int r, int i, vector<long long>& in) {
		if (l + 1 == r) return a[i] = {in[l], l, in[l], l};
		int m = (l + r)/2;
		return a[i] = build(l, m, 2*i + 1, in) + build(m, r, 2*i + 2, in);
	}

	Val decr(int l, int r, int i, int lq, int rq, long long v) {
		push(i);
		if (rq <= l || r <= lq) return a[i];
		if (lq <= l && r <= rq) {
			la[i] += v;
			push(i);
			return a[i];
		}
		int m = (l + r)/2;
		return a[i] = decr(l, m, 2*i + 1, lq, rq,v) + decr(m, r, 2*i + 2, lq, rq, v);
	}

	Val get(int l, int r, int i, int k) {
		push(i);
		if (r <= k) return {(long long)1e18, -1, -1, -1};
		if (l >= k) return a[i];
		int m = (l + r)/2;
		return get(l, m, 2*i + 1, k) + get(m, r, 2*i + 2, k);
	}

	pair<int, long long> neHi(int l, int r, int i, int k, long long hi) {
		push(i);
		if (r <= k || a[i].ma <= hi) return {-1, -1};
		if (l + 1 == r) return {l, a[i].ma};
		int m = (l + r)/2;
		auto ri = neHi(l, m, 2*i + 1, k, hi);
		if (ri.first != -1) return ri;
		return neHi(m, r, 2*i + 2, k, hi);
	}
};

std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
                                    std::vector<int> r, std::vector<int> v) {
	int n = c.size(), m = v.size();
	std::vector<int> s(n, 0);
	vector<long long> ves(m + 1);
	int ind = 0;
	for (int i = 0; i < v.size(); ++i) {
		ves[i + 1] = ves[i] + v[i];
		if (ves[i + 1] <= 0)  {
			ind = i + 1;
			ves[i + 1] = 0;
		}
	}
	SegTe se{vector<int>(4*m + 4, 0), vector<Val>(4*m + 4)};
	se.build(0, m + 1, 0, ves);
	vector<int> o(n);
	std::iota(o.begin(), o.end(),0);
	std::sort(o.begin(), o.end(), [&](int l, int r){return c[l] > c[r];});
	for (int i = 0; i < n; ++i) {
		int in = o[i];
		pair<int, long long> nex;
		while ((nex = se.neHi(0, m + 1, 0, ind, c[in])).first != -1) {
			Val neMis = se.get(0, m + 1, 0, nex.first);
			long long ch = min(nex.second - c[in], neMis.mi);
			se.decr(0, m + 1, 0, nex.first, m + 1, -ch);
			if (ch == neMis.mi) ind = neMis.miPo;
		}
		s[in] = se.get(0, m+ 1, 0, m).ma;
	}
	return s;
}

컴파일 시 표준 에러 (stderr) 메시지

candies.cpp: In member function 'void SegTe::push(int)':
candies.cpp:27:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<Val>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |    if (2*i + 2 < a.size()) {
      |        ~~~~~~~~^~~~~~~~~~
candies.cpp: In function 'std::vector<int> distribute_candies(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
candies.cpp:78:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   78 |  for (int i = 0; i < v.size(); ++i) {
      |                  ~~^~~~~~~~~~

• Subtask 1: 0 / 3
• Subtask 2: 0 / 8
• Subtask 3: 0 / 27
• Subtask 4: 0 / 29
• Subtask 5: 0 / 33