Spiraali

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <vector>
#include <string>
#include <sstream>
#define long long int int
 
int subSpiral(int y, int x, int n) {
	
	float dx = std::min(n + 1 - x, x);
	float dy = std::min(n + 1 - y, y);
	float depth = std::min(dy, dx);
 
	int m = n - (depth-1)*2;
	int firstValue = -4*depth*depth + (8 + 8 * n/2)*depth - (3 + 8 * n/2);
	int u = x - (depth-1);
	int v = y - (depth-1);
	int value = 1;
	if(v == 1 && u > 1)
	{
		value = firstValue - 1 + m*3-2 + (m-u); 
	}
	else if(u == m)
	{
		value = firstValue - 1 + m*2-1 + (m-v);
	}
	else if(v == m)
	{
		value = firstValue - 1 + m + (u-1);
	}
	else
	{
		value = firstValue - 1 + v;
	}
	return value;
}
 
int main ()
{
	int n; std::cin >> n;
	int t; std::cin >> t;
	std::vector< std::pair<int,int> > locs;
	for(int i = 0; i < t; i++) {
		int y; std::cin >> y;
		int x; std::cin >> x;
		std::pair<int, int> p = std::pair<int,int>(y,x);
		locs.push_back(p);
	}
	for(int i = 0; i < t; i++) {
		int value = subSpiral(locs[i].first, locs[i].second, n);
		std::cout << value << std::endl;
	}
}
 
 

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: WRONG ANSWER