Submission #389923

#	Time	Username	Problem	Language	Result	Execution time	Memory
389923		rainboy	Wombats (IOI13_wombats)	C11	76 / 100	20080 ms	35664 KiB

wombats

#include "wombats.h"
#include <string.h>

#define N	5000
#define M	200
#define M_	(1 << 8)	/* M_ = pow2(ceil(log2(M))) */
#define N_	(1 << 5)	/* N_ = pow2(ceil(log2(N - 1))) / M_ */

int min(int a, int b) { return a < b ? a : b; }

int aa[N][M - 1], bb[N - 1][M], n, m, st[N_ * 2][M][M];

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

	for (j1 = 0; j1 < m; j1++)
		memset(st[i][j1], 0x3f, m * sizeof *st[i][j1]);
	for (j1 = 0; j1 < m; j1++)
		for (j2 = 0; j2 < m; j2++)
			for (j3 = 0; j3 < m; j3++)
				st[i][j1][j3] = min(st[i][j1][j3], st[l][j1][j2] + st[r][j2][j3]);
}

void solve(int i_) {
	int l = i_ * M_, r = min((i_ + 1) * M_, n - 1), i, j, js;

	for (js = 0; js < m; js++) {
		int *dp = st[N_ + i_][js];

		if (l > r)
			memset(dp, 0x3f, m * sizeof *dp), dp[js] = 0;
		else {
			dp[js] = 0;
			for (j = js + 1; j < m; j++)
				dp[j] = dp[j - 1] + aa[l][j - 1];
			for (j = js - 1; j >= 0; j--)
				dp[j] = dp[j + 1] + aa[l][j];
			for (i = l + 1; i <= r; i++) {
				for (j = 0; j < m; j++)
					dp[j] += bb[i - 1][j];
				for (j = 1; j < m; j++)
					dp[j] = min(dp[j], dp[j - 1] + aa[i][j - 1]); 
				for (j = m - 2; j >= 0; j--)
					dp[j] = min(dp[j], dp[j + 1] + aa[i][j]); 
			}
		}
	}
}

void init(int n_, int m_, int A[5000][200], int B[5000][200]) {
	int i;

	n = n_, m = m_;
	for (i = 0; i < n; i++)
		memcpy(aa[i], A[i], (m - 1) * sizeof *A[i]);
	for (i = 0; i < n - 1; i++)
		memcpy(bb[i], B[i], m * sizeof *B[i]);
	for (i = 0; i < N_; i++)
		solve(i);
	for (i = N_ - 1; i > 0; i--)
		pul(i);
}

void update(int i) {
	solve(i), i += N_;
	while (i > 1)
		pul(i >>= 1);
}

void changeH(int i, int j, int w) {
	aa[i][j] = w;
	update(i / M_);
	if (i > 0 && i % M_ == 0)
		update(i / M_ - 1);
}

void changeV(int i, int j, int w) {
	bb[i][j] = w;
	update(i / M_);
}

int escape(int js, int jt) {
	return st[1][js][jt];
}

Compilation message (stderr)

grader.c: In function 'main':
grader.c:15:6: warning: variable 'res' set but not used [-Wunused-but-set-variable]
   15 |  int res;
      |      ^~~

• Subtask 1: 9 / 9
• Subtask 2: 12 / 12
• Subtask 3: 16 / 16
• Subtask 4: 18 / 18
• Subtask 5: 21 / 21
• Subtask 6: 0 / 24