Cod sursa(job #2480161)

Utilizator andrei.arnautuAndi Arnautu andrei.arnautu Data 24 octombrie 2019 23:33:08
Problema Zone Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.22 kb
/**
  *  Worg
  */
#include <map>
#include <cstdio>
#include <vector>
#include <algorithm>

FILE *fin = freopen("zone.in", "r", stdin); FILE *fout = freopen("zone.out", "w", stdout);

const int ZONE_COUNT = 9;
const int MAX_N = 500 + 1;

int n;
std::map<long long, int> exists;
std::vector<long long> zone_sums(ZONE_COUNT), ret(ZONE_COUNT);
long long sum[MAX_N][MAX_N];


void read_data() {
    scanf("%d", &n);
    for (long long& num: zone_sums) {
        scanf("%lld", &num);
        exists[num] = 1;
    }
    std::sort(zone_sums.begin(), zone_sums.end());

    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            int x; scanf("%d", &x);
            sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + x;
        }
    }
}

void make_zone_sums(int i, int j, int x, int y) {
    ret[0] = sum[i][j];
    ret[1] = sum[i][y] - ret[0];
    ret[2] = sum[i][n] - ret[1] - ret[0];

    ret[3] = sum[x][j] - ret[0];
    ret[4] = sum[x][y] - ret[3] - ret[1] - ret[0];
    ret[5] = sum[x][n] - ret[4] - ret[3] - ret[2] - ret[1] - ret[0];

    ret[6] = sum[n][j] - ret[3] - ret[0];
    ret[7] = sum[n][y] - ret[6] - ret[4] - ret[3] - ret[1] - ret[0];
    ret[8] = sum[n][n] - ret[7] - ret[6] - ret[5] - ret[4] - ret[3] - ret[2] - ret[1] - ret[0];

    std::sort(ret.begin(), ret.end());
}

std::vector<int> compute_answer() {
    for (int i = 1; i < n; i++) {
        for (int j = 1; j < n; j++) {
            if (exists[sum[i][j]] == 1) {
                for (int x = i + 1; x < n; x++) {
                    if (exists[sum[x][j] - sum[i][j]] == 1) {
                        for (int y = j + 1; y < n; y++) {
                            if (exists[sum[i][y] - sum[i][j]] == 1) {
                                make_zone_sums(i, j, x, y);
                                if (ret == zone_sums) {
                                    return {i, x, j, y};
                                }
                            }
                        }
                    }
                }
            }
        }
    }
    return {-1, -1, -1, -1};
}

int main(int argc, char *argv[]) {
    read_data();
    std::vector<int> answer = compute_answer();
    for (int& num : answer) {
        printf("%d ", num);
    }
    printf("\n");
    return 0;
}