Spiraali

#include <iostream>
#include <vector>
using namespace std;
int vali(int kohta1, int kohta2, int koko) {
    int tulos = 0, tulos2 = 0;
    if (koko % 2 == 1) {
        tulos = koko / 2 + 1 - kohta1;
        if (tulos < 0) tulos = -tulos;
        tulos2 = koko / 2 + 1 - kohta2;
        if (tulos2 < 0) tulos2 = -tulos2;
        if (tulos < tulos2) tulos = tulos2;
        if (kohta1 == koko / 2 + 1 && kohta2 == koko / 2 + 1) tulos = 0;
    }
    else {
        if (kohta1 < koko / 2) tulos = koko / 2 - kohta1;
        else if (kohta1 > koko / 2 + 1) tulos = kohta1 - (koko / 2 + 1);
        if (kohta2 < koko / 2) tulos2 = koko / 2 - kohta2;
        else if (kohta2 > koko / 2 + 1) tulos2 = kohta2 - (koko / 2 + 1);
        if (tulos < tulos2) tulos = tulos2;
    }
    return tulos;
}
long long int ekstra(int suurinKoko, int koko) {
    long long int tulos = 0;
    while (suurinKoko != koko) {
        tulos += 4LL * (suurinKoko - 1);
        suurinKoko -= 2;
    }
    return tulos;
}
long long int Yhteensa(int kohta1, int kohta2, int koko, long long int lisa) {
    long long int tulos = 0;
    if (kohta1 != koko && kohta2 == 1) {
        tulos += kohta1;
    }
    else if (kohta1 == koko && kohta2 != koko) {
        tulos += koko - 1LL + kohta2;
    }
    else if (kohta1 != 1 && kohta2 == koko) {
        tulos += 2LL * (koko - 1) + (koko - (kohta1 - 1));
    }
    else if (kohta1 == 1 && kohta2 != 1) {
        tulos += 3LL * (koko - 1) + (koko - (kohta2 - 1));
    }
    if (tulos == 0) tulos = 1;
    tulos += lisa;
    return tulos;
}
int main()
{
    int koko;
    int maara;
    cin >> koko >> maara;
    vector<int> numeroita1;
    vector<int> numeroita2;
    int num1, num2;
    for (int a = 0; a < maara; a++) {
        cin >> num1 >> num2;
        numeroita1.push_back(num1);
        numeroita2.push_back(num2);
    }
    for (int i1 = 0; i1 < maara; i1++) {
        int valiKoko = 0;
        int kohta = vali(numeroita1[i1], numeroita2[i1], koko);
        if (koko % 2 == 1) valiKoko = 1;
        else valiKoko = 2;
        valiKoko += kohta * 2;
        numeroita1[i1] -= (koko - valiKoko) / 2;
        numeroita2[i1] -= (koko - valiKoko) / 2;
        cout << Yhteensa(numeroita1[i1], numeroita2[i1], valiKoko, ekstra(koko, valiKoko)) << endl;
    }
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED