Spiraali

#include <cmath>
#include <iostream>

using namespace std;

int64_t get_layer(int64_t x, int64_t y) {
    return max(abs(x), abs(y));
}

int64_t get_offset(int64_t x, int64_t y) {
    int64_t layer = get_layer(x, y);
    int64_t offset = 4 * layer * layer - 4 * layer + 1;
    return offset;
}

int64_t get_strip_i(int64_t x, int64_t y) {
    int64_t layer = get_layer(x, y);
    int64_t width = 2 * layer + 1;
    if(x == layer) {
        return (3 * width - 3) - y + layer - 1;
    } else if(y == -layer) {
        return layer - x - 1;
    } else if(x == -layer) {
        return (width - 1) + y + layer - 1;
    } else if(y == layer) {
        return (2 * width - 2) + x + layer - 1;
    }
    return 1234;
}

int main()
{
    int64_t n, t;
    cin >> n >> t;
    //cout << get_offset(x, y) + get_strip_i(x, y) << endl;
    //cout << get_offset(x, y) << endl;
    //cout << get_strip_i(x, y) << endl;
    
    for(int64_t i = 0; i < t; i++) {
        int64_t x, y;
        cin >> y >> x;

        // Koordinaatit alkamaan (0,0)
        x--;
        y--;
    
        int64_t nx, ny;
        nx = (int64_t) (n / 2);
        ny = (int64_t) ((n - 1) / 2);
    
        x = x - nx;
        y = ny - y;

        int64_t arvo = n * n - (get_offset(x, y) + get_strip_i(x, y));
        //cout << "A " << get_strip_i(x, y) << endl;
        //cout << "B " << get_offset(x, y) << endl;
        cout << arvo << endl;
        //cout << x << " " << y << endl;
    }

    //cout << "AAAA" << endl;
    //cout << get_strip_i(0, -1) << endl;
    //cout << get_strip_i(-1, -1) << endl;
    //cout << get_strip_i(-1, 0) << endl;
    //cout << get_strip_i(-1, 1) << endl;
    //cout << get_strip_i(0, 1) << endl;
    //cout << get_strip_i(1, 1) << endl;
    //cout << get_strip_i(1, 0) << endl;
    //cout << get_strip_i(1, -1) << endl;

    //cout << "AAAA" << endl;
    //cout << get_strip_i(1, -2) << endl;
    //cout << get_strip_i(0, -2) << endl;
    //cout << get_strip_i(-1, -2) << endl;
    //cout << get_strip_i(-2, -2) << endl;
    //cout << get_strip_i(-2, -1) << endl;
    //cout << get_strip_i(-2, 0) << endl;
    //cout << get_strip_i(-2, 1) << endl;
    //cout << get_strip_i(-2, 2) << endl;
    //cout << get_strip_i(-1, 2) << endl;
    //cout << get_strip_i(0, 2) << endl;
    //cout << get_strip_i(1, 2) << endl;
    //cout << get_strip_i(2, 2) << endl;
    //cout << get_strip_i(2, 1) << endl;
    //cout << get_strip_i(2, 0) << endl;
    //cout << get_strip_i(2, -1) << endl;
    //cout << get_strip_i(2, -2) << endl;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED