Spiraali

#include <bits/stdc++.h>
using namespace std;
 
int getNum(int n, int x, int y) {
	if(x == 1) {
		return y;
	}
	else {
		int c = 0, cx, cy;
		int tcx, tcy;
		tcx = min(n-x, x);
		tcy = min(n-y, y);
		for(int i = 1; true; i++) {
 
			c = i*(2*n-i);
			cx = i/2;
			cy = (i-1)/2;
 
			if(i % 2 == 1) {
				if(cx == tcx) {
					return ((tcx*2+1)*2*n- (tcx*2+1)*(tcx*2+1) + n - y - (tcx-1/2));
				}
				if(cy == tcy) {
					return ((tcy*2+1)*(2*n-(tcy*2+1)) - n + (x + (tcy*2+1)/2));
				}
			}
			else {
				if(cx + 1 == x) {
					return (c - cy + y - 1);
				}
				if(cy + 1 == y) {
					return (c - x + cx + 1);
				}
			};
		}
	}
}
 
int main () {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	
	int n, t, x, y;
	cin >> n >> t;
	for(int i = 0; i < t; i++) {
		cin >> y >> x;
		cout << getNum(n, x, y) << "\n";
	}
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED