洛谷 P1141 01迷宫 【一道并查集巧解的题】

Lg P1141 01迷宫

思路

不难想到，该题可以用dfs做
但是这道题用并查集的方法可能还更简单些
因为从一个点出发只能走到与他数字不同的那个点，所以单独在并查集模板的基础上用一个num数组表示当前节点及其子节点的连通块数。

代码实现及玄学优化

1.因为当前这个点只与上一个点有关,所以可以用一个滚动数组优化内存
2.如何实现从一个点出发只能走到与他数字不同的那个点？。对每一个点来说，如果他和他上面和左边的权值不一样，那么就可以将这两个点合并。并且维护处num值。为什么不判断上下左右4个点的值，而只判断上左的值？下与右会在接下来的判断中判断。所以只用判断上，左的值。

CODE

#include <bits/stdc++.h>
using namespace std;
char s[2][1005];
int f[1000005],num[1000005];
int n,m;
void init()
{
	freopen("Lg 1141.in","r",stdin);
}

int find(int x){return f[x]==x?x:f[x]=find(f[x]);}

int id(int i,int j){return i*n+j;}

void merge(int x,int y)
{
	int fa1=find(x),fa2=find(y);
	if(fa1!=fa2)
	{
		f[fa2]=fa1;
		num[fa1]+=num[fa2];
	}

}

void readdata()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	{
		scanf("%s",s[i&1]);
		for(int j=0;j<n;j++)
		{
			int now=id(i,j);
			f[now]=now;num[now]=1;
			if(i!=0 && s[(i-1)&1][j]!=s[i&1][j])merge(id(i-1,j),now);
			if(j!=0 && s[i&1][j-1]!=s[i&1][j])merge(id(i,j-1),now);
		}
	}
	for(int i=1;i<=m;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		printf("%d\n",num[find(id(x-1,y-1))]);
	}
}

int main()
{
//	init();
	readdata();
//	work();
	return 0;
}