【bzoj5101】[POI2018]Powód

Description

在地面上有一个水箱，它的俯视图被划分成了n行m列个方格，相邻两个方格之间有一堵厚度可以忽略不计的墙，水
箱与外界之间有一堵高度无穷大的墙，因此水不可能漏到外面。已知水箱内每个格子的高度都是[0,H]之间的整数
，请统计有多少可能的水位情况。因为答案可能很大，请对10^9+7取模输出。两个情况不同当且仅当存在至少一个
方格的水位在两个情况中不同。
Input

第一行包含三个正整数n,m,H(n*m<=500000,1<=H<=10^9)。
接下来n行，每行m-1个整数a[i]j，表示(i,j)和(i,j+1)之间的墙的高度。
接下来n-1行，每行m个整数b[i]j，表示(i,j)和(i+1,j)之间的墙的高度。
Output

输出一行一个整数，即方案数模10^9+7的结果。
Sample Input

3 2 2

1

1

1

1 2

1 1
Sample Output

65

HINT

要么全部格子水位都是2，要么全部格子水位都在[0,1]之间，共1+2^6=65种情况。

题解
把格子当成点，墙当成边，做最小生成树。
方案g=(g1+now-h1)*(g2+now-h2)

代码

#include<bits/stdc++.h>
typedef long long ll;
const int mod=1000000007;
const int N=500010;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct node{int a,b,l;}e[N<<1];
int n,m,H,ans,f[N],g[N],h[N],tot;
inline bool cmp(node a,node b){return a.l<b.l;}
int find(int i){if (f[i]!=i) f[i]=find(f[i]);return f[i];}
int main()
{
    n=read();m=read();H=read();
    for (int i=1;i<=n;i++)
        for (int j=1;j<m;j++)
        {
            e[++tot].a=(i-1)*m+j;e[tot].b=(i-1)*m+j+1;
            e[tot].l=read();
        }
    for (int i=1;i<n;i++)
        for (int j=1;j<=m;j++)
        {
            e[++tot].a=(i-1)*m+j;e[tot].b=i*m+j;
            e[tot].l=read();
        }
    sort(e+1,e+tot+1,cmp);for (int i=1;i<=n*m;i++) g[i]=1,f[i]=i,h[i]=0;
    for (int i=1;i<=tot;i++)
    {
        int x=find(e[i].a),y=find(e[i].b);
        if (x!=y)
        {
            f[y]=x;
            g[x]=(ll)(g[x]+e[i].l-h[x])*(g[y]+e[i].l-h[y])%mod;
            h[x]=e[i].l;
        }
    }
    printf("%d",(g[find(1)]+H-h[find(1)])%mod);
}