Cod sursa(job #1082992)

Utilizator WyvernFMI Stanescu Leonard Wyvern Data 15 ianuarie 2014 14:47:01
Problema Matrice 2 Scor 100
Compilator cpp Status done
Runda Teme Pregatire ACM Unibuc 2013 Marime 1.98 kb
#include <cstdio>
#include <algorithm>
#include <fstream>
using namespace std;
ifstream fi("matrice2.in");
ofstream fo("matrice2.out");
int v1[4]={0,0,1,-1},v2[4]={1,-1,0,0},n,m,l,u[310][310],ind[310][310],t[90010];
struct elem {int val,x,y;}v[90010];
struct query {int x1, y1, x2, y2, sol, ind;}q[20010];
int cmp(elem a, elem b) {return a.val>b.val;}
int cmp2(query a, query b){return a.sol > b.sol;}
int cmp3(query a, query b){return a.ind < b.ind;}
int root (int x) {if (x!=t[x]) t[x]=root(t[x]);return t[x];}
int main() {
    fi>>n>>m;
    int i,j,h=0,x,y,cx,cy,step,p,k,pos;
    for (i=1; i<=n; i++)
        for (j=1; j<=n; j++) {
            fi>>v[++h].val;
            v[h].x=i;
            v[h].y=j;
        }
    l=n*n;
    sort (v+1, v+l+1, cmp);
    for (i=1; i<=l; i++)
        ind[v[i].x][v[i].y]=i;
    for (i=1; i<=m; i++) {
        fi>>q[i].x1>>q[i].y1>>q[i].x2>>q[i].y2;
        q[i].ind=i;
    }
    for (step=1;step<v[1].val;step<<=1);
    for (;step>0;step>>=1) {
        sort (q+1,q+m+1,cmp2);
        for (i=1;i<=n;i++)
            for (j=1;j<=n;j++) u[i][j]=0;
        for (i=1;i<=l;i++) t[i]=i;
        j=1;
        for (i=1;i<=l;) {
            for (; j<=m && q[j].sol+step>v[i].val; j++)
                if (root(ind[q[j].x1][q[j].y1])==root(ind[q[j].x2][q[j].y2])) q[j].sol+=step;
            pos=i;
            for (;i<=l&&v[i].val==v[pos].val;i++) {
                x=v[i].x;
                y=v[i].y;
                u[x][y]=1;
                for (k=0;k<4;k++) {
                    cx = x + v1[k];
                    cy = y + v2[k];
                    if (cx>0&&cy>0&&cx<=n&&cy<=n&&u[cx][cy]) {
                        p=ind[cx][cy];
                        t[root(p)]=root(i);
                    }
                }
            }
        }
        for (;j<=m;j++)
            if (root(ind[q[j].x1][q[j].y1])==root(ind[q[j].x2][q[j].y2])) q[j].sol+=step;
    }
    sort(q+1,q+m+1,cmp3);
    for (i=1; i<=m; i++) fo<<q[i].sol<<'\n';
}