Spiraali

#include <iostream>
#include <cmath>

using namespace std;

long long int find_ans(int y, int x){
    long long int ind = 0;
    if(x * x >= y*y){
        ind = 4 * x * x + x + y + 1;
        if(x > y){
            ind += 2 * (x-y);
        }
        }
    else{
        ind = 4 * y * y + x + y +1;
        if(x > y){
            ind -= 2 * (x-y);
        }
    }


    return ind;
}

int main()
{
    int n, t; cin >> n >> t;
    while(t--){
        int y, x; cin >> y >> x;
        x = x - (n/2+1);
        if(y == n/2)y = 0;
        else if(y < n/2){
            y = -1 * (y-n/2);
        }
        else{
            y = -1 * (y-n/2);
        }
        cout << n*n - find_ans(y, x) +1 << "\n";
    }









    return 0;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: WRONG ANSWER