답안 #871117

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
871117 2023-11-10T00:01:39 Z rainboy Santa Claus (RMI19_santa) C
100 / 100
221 ms 6700 KB
#include <assert.h>
#include <stdio.h>
#include <string.h>

#define N	100000
#define N_	(1 << 17)	/* N_ = pow2(ceil(log2(N))) */

int max(int a, int b) { return a > b ? a : b; }

int n_;

int ss[N_ * 2], pp[N_ * 2];

void pul(int i) {
	int l = i << 1, r = l | 1;

	ss[i] = ss[l] + ss[r];
	pp[i] = max(pp[l], ss[l] + pp[r]);
}

void update(int i, int x) {
	i += n_;
	pp[i] = max(ss[i] += x, 0);
	while (i > 1)
		pul(i >>= 1);
}

int ss_[N_ * 2];

void update_(int i, int x) {
	for (i += n_; i > 0; i >>= 1)
		ss_[i] += x;
}

int query_(int l) {
	int r = n_ - 1;

	for (l += n_, r += n_; l <= r; l >>= 1, r >>= 1)
		if ((l & 1) == 1) {
			if (ss_[l]) {
				while (l < n_)
					l = ss_[l << 1 | 0] ? l << 1 | 0 : l << 1 | 1;
				return l - n_;
			}
			l++;
		}
	return -1;
}

int main() {
	int t;

	scanf("%d", &t);
	while (t--) {
		static int xx[N], aa[N], tt[N];
		int n, i, j, k, a;

		scanf("%d", &n);
		assert(n <= 99999);
		for (i = 0; i < n; i++)
			scanf("%d", &xx[i]);
		k = -1;
		for (i = 0; i < n; i++) {
			scanf("%d", &tt[i]);
			if (tt[i] == 0)
				k = i;
		}
		for (i = 0; i < n; i++)
			scanf("%d", &aa[i]);
		n_ = 1;
		while (n_ < n + 1)
			n_ <<= 1;
		memset(ss, 0, n_ * 2 * sizeof *ss), memset(pp, 0, n_ * 2 * sizeof *pp);
		memset(ss_, 0, n_ * 2 * sizeof *ss_);
		for (i = 0, j = 0; j < n; j++) {
			update(aa[j], tt[j] == 0 ? 1 : -1);
			if (j >= k)
				while (i <= j && pp[1] == 0) {
					if (tt[i] == 0)
						update_(aa[i], 1);
					else {
						update(aa[i], 1);
						if ((a = query_(aa[i])) != -1)
							update(a, -1), update_(a, -1);
					}
					i++;
				}
			printf("%d ", i == 0 ? -1 : xx[j] * 2 - xx[i - 1]);
		}
		printf("\n");
	}
	return 0;
}

Compilation message

santa.c: In function 'main':
santa.c:53:2: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   53 |  scanf("%d", &t);
      |  ^~~~~~~~~~~~~~~
santa.c:58:3: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   58 |   scanf("%d", &n);
      |   ^~~~~~~~~~~~~~~
santa.c:61:4: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   61 |    scanf("%d", &xx[i]);
      |    ^~~~~~~~~~~~~~~~~~~
santa.c:64:4: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   64 |    scanf("%d", &tt[i]);
      |    ^~~~~~~~~~~~~~~~~~~
santa.c:69:4: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   69 |    scanf("%d", &aa[i]);
      |    ^~~~~~~~~~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 2396 KB Output is correct
2 Correct 1 ms 2396 KB Output is correct
3 Correct 5 ms 2396 KB Output is correct
4 Correct 11 ms 2584 KB Output is correct
5 Correct 25 ms 2856 KB Output is correct
6 Correct 40 ms 3028 KB Output is correct
7 Correct 77 ms 3672 KB Output is correct
8 Correct 126 ms 4620 KB Output is correct
9 Correct 159 ms 5972 KB Output is correct
10 Correct 221 ms 6700 KB Output is correct