Submission #389942

#TimeUsernameProblemLanguageResultExecution timeMemory
389942rainboyWombats (IOI13_wombats)C11
76 / 100
20025 ms56276 KiB
#include "wombats.h"
#include <string.h>
 
#define N	5000
#define M	200
#define M_	40
#define N_	(1 << 7)	/* N_ = pow2(ceil(log2(N - 1))) / M_ */
#define INF	0x3f3f3f3f
 
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, x;
	int (*sti)[M] = st[i], (*stl)[M] = st[l], (*str)[M] = st[r];
 
	for (j1 = 0; j1 < m; j1++)
		memset(sti[j1], 0x3f, m * sizeof *sti[j1]);
	for (j1 = 0; j1 < m; j1++) {
		int *sti1 = sti[j1], *stl1 = stl[j1];
 
		for (j2 = 0; j2 < m; j2++) {
			int *str2 = str[j2], a = stl1[j2];
 
			if (a == INF)
				continue;
			for (j3 = 0; j3 < m; j3++)
				if (sti1[j3] > (x = a + str2[j3]))
					sti1[j3] = x;
		}
	}
}
 
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], x;
 
		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];
					if ((i < r || i == n - 1) && j > 0 && dp[j] > (x = dp[j - 1] + aa[i][j - 1]))
						dp[j] = x;
				}
				if (i < r || i == n - 1)
					for (j = m - 2; j >= 0; j--)
						if (dp[j] > (x = dp[j + 1] + aa[i][j]))
							dp[j] = x;
			}
		}
	}
}
 
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_ == 0 && i == n - 1 ? i / M_ - 1 : i / M_);
}
 
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;
      |      ^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...