Submission #1067829

#	Time	Username	Problem	Language	Result	Execution time	Memory
1067829		sleepntsheep	Wombats (IOI13_wombats)	C++17	9 / 100	614 ms	11344 KiB

wombats

#include "wombats.h"
#include <cstdio>
#include <set>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;

template <typename T>
using ve = vector<T>;


int r, c, at[9999], h[5000][200], v[5000][200];
int sum = 0;

int ans_5[100][100];

void calc_sub5_dp() {
	if (c == 1) return ;
	static int dp5[5000][100];


	for (int i = 0; i < r; ++i) {
	}

	for (int ii = 0; ii < c; ++ii) {
		/* find answer for (V1, V2) where V1 = ii */
		memset(dp5, 63, sizeof dp5);
		dp5[0][ii] = 0;

		for (int i = 0; i < r; ++i) {
			if (i > 1)
				for (int j = 0; j < c; ++j)
					dp5[i][j] = dp5[i - 1][j] + v[i - 1][j];

			static int ph[102];
			ph[0] = 0;
			for (int j = 0; j + 1 < c; ++j)
				ph[j + 1] = ph[j] + h[i][j];

			static int pre[102], suf[102]; /* annoying ahh math */ 

			pre[0] = dp5[i][0]; /* - ph[0] = 0; */
			for (int j = 1; j < c; ++j)
				pre[i] = min(pre[i - 1], dp5[i][j] - ph[j]);
			suf[c - 1] = dp5[i][c - 1] + ph[c - 1];
			for (int j = c - 2; j >= 0; --j)
				suf[i] = min(suf[i + 1], dp5[i][j] + ph[j]);

			for (int j = 0; j < c; ++j)
				dp5[i][j] = min({dp5[i][j], ph[j] + pre[j], suf[j] - ph[j]});
		}
		for (int jj = 0; jj < c; ++jj) ans_5[ii][jj] = dp5[r - 1][jj];
	}
}

void changeH(int P, int Q, int W) {
	if (c <= 100) {
		h[P][Q] = W;
		calc_sub5_dp();
	}
}

void changeV(int P, int Q, int W) {
	if (c == 1) {
		sum += W - at[P];
		at[P] = W;
	} else if (c <= 100) {
		v[P][Q] = W;
		calc_sub5_dp();
	}
}


void init(int R, int C, int H[5000][200], int V[5000][200]) {
	r = R, c = C;

	if (c == 1)
		for (int i = 0; i < r; ++i) changeV(i, 0, V[i][0]);

	if (c <= 100) {
		calc_sub5_dp();
	}

}

int escape(int V1, int V2) {
	if (c == 1)
		return sum;
	else if (c <= 100) {
		return ans_5[V1][V2];
	}
	return 0;
}

Compilation message (stderr)

grader.c: In function 'int 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: 0 / 12
• Subtask 3: 0 / 16
• Subtask 4: 0 / 18
• Subtask 5: 0 / 21
• Subtask 6: 0 / 24