Cod sursa(job #1392985)

Utilizator retrogradLucian Bicsi retrograd Data 19 martie 2015 00:19:38
Problema Matrice 2 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.68 kb
#include<fstream>
#include<vector>
#include<algorithm>

using namespace std;
typedef int var;

ifstream fin("matrice2.in");
ofstream fout("matrice2.out");

struct Edge {
    var a, b, c;
    Edge(var x, var y, var z) {
        a = x;
        b = y;
        c = z;
    }
    const bool operator<(const Edge &e) const{
        return c > e.c;
    }
};

#define MAXN 305
#define MAX_NO 305*305*2

var n, m;
var A[MAXN][MAXN];
var PARENT[MAX_NO], D[MAX_NO], COST[MAX_NO];;
vector<Edge> E;

var Parent[MAXN*MAXN], Rank[MAXN*MAXN], Node[MAXN*MAXN];
var Find(var f) {
    if(!Parent[f]) return f;
    Parent[f] = Find(Parent[f]);
    return Parent[f];
}
void Union(var r1, var r2, var under) {
    r1 = Find(r1);
    r2 = Find(r2);
    if(Rank[r1] < Rank[r2]) {
        Parent[r1] = r2;
        Node[r2] = under;
    } else {
        Parent[r2] = r1;
        Node[r1] = under;
        Rank[r1] += (Rank[r1] == Rank[r2]);
    }
}

var toNod(var i, var j) {
    return n*(i-1)+j;
}

var lca(var n1, var n2) {
    while(D[n1] > D[n2]) n1 = PARENT[n1];
    while(D[n2] > D[n1]) n2 = PARENT[n2];

    while(n1 != n2) {n1 = PARENT[n1]; n2 = PARENT[n2];}
    return n1;
}

int main() {

    fin>>n>>m;
    for(var i=1; i<=n; i++) {
        for(var j=1; j<=n; j++) {
            fin>>A[i][j];
        }
    }
/*
    for(var i=1; i<=n; i++) {
        for(var j=1; j<=n; j++) {
            var t = toNod(i, j);

            var i1 = (t - 1) / n + 1;
            var j1 = (t - 1) % n + 1;

            fout<<i1<<" "<<j1<<" ; ";
        }
        fout<<endl;
    }
*/
    for(var i=1; i<n; i++) {
        for(var j=1; j<n; j++) {
            E.push_back(Edge(toNod(i, j), toNod(i+1, j), min(A[i][j], A[i+1][j])));
            E.push_back(Edge(toNod(i, j), toNod(i, j+1), min(A[i][j], A[i][j+1])));
        }
    }
    for(var i=1; i<n; i++) {
        E.push_back(Edge(toNod(n, i), toNod(n, i+1), min(A[n][i], A[n][i+1])));
        E.push_back(Edge(toNod(i, n), toNod(i+1, n), min(A[i][n], A[i+1][n])));
    }

    sort(E.begin(), E.end());
    var nodes = n*n;

    for(var i=1; i<=nodes; i++) {
        Node[i] = i;
    }

    for(auto e : E) {
        if(Find(e.a) != Find(e.b)) {
            nodes ++;

            PARENT[Node[Find(e.a)]] = nodes;
            PARENT[Node[Find(e.b)]] = nodes;

            COST[nodes] = e.c;

            Union(e.a, e.b, nodes);
        }
    }

    for(var i=nodes; i>=1; i--) {
        D[i] = D[PARENT[i]] + 1;
    }

    var a, b, c, d, u, v;

    while(m--) {
        fin>>a>>b>>c>>d;
        u = toNod(a, b);
        v = toNod(c, d);

        fout<<COST[lca(u, v)]<<'\n';
    }

    return 0;
}