bzoj 5101: [POI2018]Powód 最小生成树+并查集

题意

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

分析

考虑把每堵墙看做一条边，不难发现只有最小生成树上的边是有用的。
那么我们考虑在Kruskal算法的加边过程中顺便统计答案。
对于当前每一个连通块x，设mx[x]表示x中的最大边权，s[x]表示在连通块x中每个点的点权不超过mx[x]时的方案数。
当合并两个连通块x和y时，设当前边权为w，那么显然有s[x+y]=(s[x]+w-mx[x])*(s[y]+w-mx[y])。
就做完了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=500005;
const int MOD=1000000007;

int n,m,h,cnt,f[N],s[N],mx[N];
struct edge{int x,y,w;}e[N*2];

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;
}

int point(int x,int y)
{
    return (x-1)*m+y;
}

bool cmp(edge a,edge b)
{
    return a.w<b.w;
}

int find(int x)
{
    if (f[x]==x) return x;
    else return f[x]=find(f[x]);
}

int main()
{
    n=read();m=read();h=read();
    for (int i=1;i<=n;i++)
        for (int j=1;j<m;j++)
            e[++cnt].x=point(i,j),e[cnt].y=point(i,j+1),e[cnt].w=read();
    for (int i=1;i<n;i++)
        for (int j=1;j<=m;j++)
            e[++cnt].x=point(i,j),e[cnt].y=point(i+1,j),e[cnt].w=read();
    for (int i=1;i<=n*m;i++) f[i]=i,s[i]=1,mx[i]=0;
    sort(e+1,e+cnt+1,cmp);
    for (int i=1;i<=cnt;i++)
    {
        int x=find(e[i].x),y=find(e[i].y),w=e[i].w;
        if (x==y) continue;
        s[y]=(LL)(s[x]+w-mx[x])*(s[y]+w-mx[y])%MOD;f[x]=y;mx[y]=w;
    }
    printf("%d",(s[find(1)]+h-mx[find(1)])%MOD);
    return 0;
}