Ruudukko

#include <algorithm>
#include <cassert>
#include <cmath>
#include <iostream>
#include <numeric>
#include <vector>
 
using ll = long long;
 
class NxNArray {
public:
    NxNArray(int n_): n(n_) { array.resize(n * n, 0); }
 
    ll& operator()(int row, int col) { return array[row * n + col]; }
    ll& operator[](int i) { return array[i]; }
 
    int size() const { return n; }
 
    ll sum_at(int row, int col) {
        ll sum = 0;
        for (int r = 0; r < n; ++r) { sum += (*this)(r, col); }
        for (int c = 0; c < n; ++c) { sum += (*this)(row, c); }
        // Älä laske itseään
        sum -= 2 * (*this)(row, col);
        // Reitti voi myös loppua tähän
        sum += 1;
        return sum;
    }
    ll sum_at(int i) { return sum_at(i / n, i % n); }
 
    int n;
    std::vector<ll> array;
};
 
NxNArray read_input() {
    int n;
    std::cin >> n;
    NxNArray array{ n };
 
    for (int i = 0; i < n * n; ++i) {
        int v;
        std::cin >> v;
        array[i] = v;
    }
    return array;
}
 
int main() {
    std::cin.sync_with_stdio(false);
 
    auto array = read_input();
    //    array.array = { 2, 1, 3, 1, 1, 1, 9, 2, 7 };
 
    NxNArray sums(array.size());
 
    const int n2 = array.size() * array.size();
    for (int smallest = 0; smallest < n2; ++smallest) {
        // Lasketaan summa kohdissa, joissa on pienin tähän laskematon luku
        // Tehdään summien listasta kopio, jottei yhtä suuria lukuja lasketa mukaan
        auto tmp_sums = sums;
        for (int i = 0; i < n2; ++i) {
            if (array[i] == smallest + 1) { tmp_sums[i] = sums.sum_at(i); }
        }
        sums = std::move(tmp_sums);
    }
    // Lopussa lasketaan kaikki yhteen
    ll res = std::reduce(sums.array.begin(), sums.array.end());
    std::cout << res;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED