Description
在地面上有一个水箱,它的俯视图被划分成了n行m列个方格,相邻两个方格之间有一堵厚度可以忽略不计的墙,水 箱与外界之间有一堵高度无穷大的墙,因此水不可能漏到外面。已知水箱内每个格子的高度都是[0,H]之间的整数 ,请统计有多少可能的水位情况。因为答案可能很大,请对10^9+7取模输出。两个情况不同当且仅当存在至少一个 方格的水位在两个情况中不同。
第一行包含三个正整数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的结果。
3 2 2
1
1
1
1 2
1 1
Sample Output
65
HINT
先吐槽一波,你边界围墙高度既然是无穷大,那么也就是说我填多少水都可以咯,那方案数不应该是无穷大吗,enmmm 考虑把每堵墙看做一条边,不难发现只有最小生成树上的边是有用的。 那么我们考虑在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])。 就做完了。
CODE
#include <bits/stdc++.h>
typedef long long ll;
const int N =
500005;
const int MOD =
1000000007;
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 Edge
{
int x,y,w;
}e[N *
2];
int cnt;
int n,m,h;
int point(
int x,
int y)
{
return (x -
1) * m + y;
}
bool cmp(Edge a,Edge b)
{
return a.w < b.w;
}
int f[N],s[N],mx[N];
int find(
int x)
{
return f[x] == x ? x : 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;
std::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] =
1ll * (s[x] + w - mx[x]) * (s[y] + w - mx[y]) % MOD;
f[x] = y, mx[y] = w;
}
printf(
"%d\n",(s[find(
1)] + h - mx[find(
1)]) % MOD);
}