LOJ3246 「USACO 2020.1 Platinum」Cave Paintings

Link
用并查集把每一行在同一个连通块内的点并起来,并向下一行的与当前连通块有交的连通块连边。
这样会构成一棵森林的结构,然后直接dp即可。

#include <cstdio>
using i64=long long;
const int N=1007,P=1000000007;
int id[N][N],fa[N*N];i64 f[N*N];char str[N][N];
int find(int x){return x==fa[x]? x:fa[x]=find(fa[x]);}
int main()
{
    int n,m,cnt=0;i64 ans=1;scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%s",str[i]+1);
    for(int i=n-1;i>1;--i)
    {
	int now=cnt;
	for(int j=2,k;j<m;++j)
	    if(str[i][j]=='.')
	    {
		if(str[i][j-1]=='#') f[++cnt]=1,fa[cnt]=cnt;
		if(id[i][j]=cnt,str[i+1][j]=='.') if((k=find(id[i+1][j]))!=cnt) (f[cnt]*=f[k])%=P,fa[k]=cnt;
	    }
	for(int j=now+1;j<=cnt;++j) if(find(j)==j) ++f[j];
    }
    for(int i=1;i<=cnt;++i) if(find(i)==i) (ans*=f[i])%=P;
    printf("%lld",ans);
}
posted @ 2020-06-12 08:14  Shiina_Mashiro  阅读(245)  评论(0编辑  收藏  举报