Spiraali

#include <math.h>
#include <iostream>
#include <string>
#include <vector>
using namespace std;

int k(int x, int y, int n);

int main()
{
	int n, t, x, y, i;
	vector<long long> v;
	cin >> n >> t;

	for (i = 0; i < t; i++) {
		cin >> x >> y;
		v.push_back(k(x, y, n));
	}

	for (auto i : v)
	{
		cout << i << "\n";
	}
}

int k(int x, int y, int n) {
	if (x == n / 2 && y == n / 2 + 1) return n*n;

	if (x < y) {
		int k = n - x + 1;
		if (y <= k) {
			long long l = 0;
			if (x > 1) l += ((x-1)*((4 * n - 4 + (4 * (n - 2 * (x - 2)) - 4))))/2;
			l += n - 2 * (x - 1);
			l += 2 * (n - 2 * (x - 1) - 1);
			l += n - y - (x - 1);
			return l;
		}
		else
		{
			k = n - y;
			long long l = 0;
			l += ((k)*(4 * n - 4 + (4 * (n - 2 * (k - 1)) - 4))) / 2;
			l += n - 2 * k;
			l += n - 2 * k - 1;
			l += (n - 2 * k) - (x - k);
			return l;
		}
	}
	else if (x == y) {
		if (x <= n / 2)
		{
			long long l = 1;
			l += ((x - 1) * ((4 * n - 4 + (4 * (n - 2 * (x - 2)) - 4)))) / 2;
			return l;
		}
		else
		{
			int k = n - x;
			long long l = 0;
			l += ((k) * ((4 * n - 4 + (4 * (n - 2 * (k - 1)) - 4)))) / 2;
			l += 2 * (n - 2 * k);
			return l - 1;
		}
	}
	else 
	{
		int k = n - y + 1;
		if (x <= k) {
			long long l = 0;
			if (x > 1) l += ((y - 1)*((4 * n - 4 + (4 * (n - 2 * (y - 2)) - 4)))) / 2;
			l += x - (y - 1);
			return l;
		}
		else
		{
			k = n - x;
			long long l = 0;
			l += (k*((4 * n - 4 + (4 * (n - 2 * (k - 1)) - 4)))) / 2;
			l += n - 2 * k - 1;
			l += y - k;
			return l;
		}
	}
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: WRONG ANSWER