【JZOJ A组】water

Description

有一块矩形土地被划分成 n*m 个正方形小块。这些小块高低不 平，每一小块都有自己的高度。水流可以由任意一块地流向周围四个 方向的四块地中，但是不能直接流入对角相连的小块中。 一场大雨后，由于地势高低不同，许多地方都积存了不少降水。 给定每个小块的高度，求每个小块的积水高度。 注意：假设矩形地外围无限大且高度为 0。

Input

第一行包含两个非负整数 n,m。 接下来 n 行每行 m 个整数表示第 i 行第 j 列的小块的高度。

Output

输出 n 行，每行 m 个由空格隔开的非负整数，表示每个小块的积 水高度。

Sample Input

3 3
4 4 0
2 1 3
3 3 -1

Sample Output

0 0 0
0 1 0
0 0 1

Data Constraint

对于 20%的数据 n,m<=4
对于 40%的数据 n,m<=15
对于 60%的数据 n,m<=50
对于 100%的数据 n,m<=300，|小块高度|<=10^9。
在每一部分数据中，均有一半数据保证小块高度非负

思路

一个块的高度就是从这个块走出矩形的所有路径上的最大值的最小值。
相邻块连边，权值为两块的较大值，矩形边界的块向“矩形外”连边，权值为max(高度，0)
做最小生成树。
每个水的高度是节点到根路径上的max值

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=377,inf=0x3f3f3f3f;
struct C
{
    int u,v,val;
}c[maxn*maxn*10];
struct E
{
    int to,next,v;
}e[maxn*maxn*10];
int a[maxn][maxn],n,m,f[maxn*maxn],cnt,list[maxn*maxn],t,b[maxn][maxn];
int cmp(C x,C y)
{
    return x.val<y.val;
}
int gf(int x)
{
    return x==f[x]?x:f[x]=gf(f[x]);
}
void add(int u,int v,int val)
{
    e[++cnt].to=v; e[cnt].next=list[u]; list[u]=cnt; e[cnt].v=val;
//  printf("u=%d v=%d val=%d\n",u,v,val);
}
void doit(int x1,int y1,int x,int y)
{
    int v1,v;
    if(!x1||x1>n||!y1||y1>m) v1=0; else v1=x1*m+y1-m;
    if(!x||x>n||!y||y>m) v=0; else v=x*m+y-m;
    if(!v1&&!v) return;
/*  add(v,v1,max(a[x][y],a[x1][y1]));
    add(v1,v,max(a[x][y],a[x1][y1]));*/
    c[++t].u=v; c[t].v=v1; c[t].val=max(a[x][y],a[x1][y1]);
    c[++t].u=v1; c[t].v=v; c[t].val=max(a[x][y],a[x1][y1]);

}
void mintree()
{
    for(int i=1; i<=n*m; i++) f[i]=i;
    for(int i=1; i<=t; i++)
    {
        int u=gf(c[i].u),v=gf(c[i].v);
        if(u!=v) f[u]=v,add(c[i].u,c[i].v,c[i].val),add(c[i].v,c[i].u,c[i].val);
    }
}
void dfs(int u,int fa,int x)
{
    for(int i=list[u]; i; i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa) continue;
        int px=v/m+1,py=v%m;
        if(!py) px--,py=m;
        a[px][py]=max(x,e[i].v)-a[px][py];
        b[px][py]=1;
        dfs(v,u,max(x,e[i].v));
    }
}
int main()
{
     freopen("water.in","r",stdin);
     freopen("water.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++) for(int j=1; j<=m; j++) scanf("%d",&a[i][j]);
    for(int i=1; i<=n; i++) for(int j=1; j<=m; j++)
    {
        doit(i+1,j,i,j); doit(i-1,j,i,j);
        doit(i,j+1,i,j); doit(i,j-1,i,j);
    }
    sort(c+1,c+t+1,cmp);
    mintree();
    dfs(0,-1,-inf);
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++) printf("%d ",a[i][j]);
        printf("\n");
    }
}