Pallo

#include <iostream>
#include <string>
#include <math.h>
#include <map>
#include <algorithm>
#include <vector>
#include <fstream>
#include <set>
#include <unordered_map>
#include <queue>
#include <climits>
#include <stack>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define LOOP(i,a,l) for (int i = a; i < l; i++)
#define LUO_VERKKO LOOP(i,0,m) {int a,b; cin >> a >> b; v[a].push_back(b); v[b].push_back(a);}
#define COLLECT for (int i = 0; i < n; i++) cin >> s[i]
#define PRINT_LIST(l) for (int i = 0; i < l.size(); i++) cout << l[i] << " "
#define S second
#define F first
#define A 999999999999999999LL
#define M 1000000007
//https://cses.fi/dt/
int n;
int t;
#define N (1<<17)
pair<int,int> d[1010][1010];
int main() {
	cin >> t;
	for (int i = 0; i < t; i++)
	{
		int n, m, k; cin >> n >> m >> k;
		int x = 1;
		int y = 1;
		int c = 0;
		int dir = 3;
		int moves = 0;
		for (int l = 0; l < m; l++)
		{
			for (int o = 0; o < n; o++)
			{
				d[l][o] = { 0,0 };
			}
		}
		while (k > 0) {
			if (d[x][y].first) {
				while (d[x][y].first <= k) {
					k -= d[x][y].first;
					c += d[x][y].second;
				}
			}
			if (x == 0 || y == 0 || x == m - 1 || y == n - 1) c++;
			if (x == 0) {
				if (dir == 2) dir = 3;
				if (dir == 1) dir = 4;
			}
			if (y == 0) {
				if (dir == 1) dir = 2;
				if (dir == 4) dir = 3;
			}
			if (x == m-1) {
				if (dir == 3) dir = 2;
				if (dir == 4) dir = 1;
			}
			if (y == n-1) {
				if (dir == 2) dir = 1;
				if (dir == 3) dir = 4;
			}
			d[x][y] = { moves,c };
			k--;
			if (dir == 1) {
				x--;
				y--;
			}
			if (dir == 2) {
				x--;
				y++;
			}
			if (dir == 3) {
				x++;
				y++;
			}
			if (dir == 4) {
				x++;
				y--;
			}
			moves++;
		}
		cout << c <<"\n";
	}
}

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: RUNTIME ERROR