Spiraali

#include <iostream>
#include <vector>
using namespace std;
int vali(int kohta1, int kohta2, int koko) {
    int tulos = 0, tulos2 = 0;
    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(long long int suurinKoko, int koko, int kohta) {
    long long int tulos = 0;
    if (suurinKoko - koko != 0) {
        int dif = (suurinKoko + 2) / 2 - 1;
        int dif2 = koko / 2 - 1;
        tulos = dif * 4LL * dif - dif2 * 4LL * dif2 - 4LL * (koko - 1);
    }
    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);
        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, kohta)) << endl;
    }
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED