The 2019 ICPC China Shaanxi Provincial Programming Contest B. Grid with Arrows

赛时没写出来的一道，当时想着用并查集来维护是否联通了，但是开了个结构体的二维数组，空间限制超了，后来看题解发现可以用一维代替二维。

题目链接：Problem - B - Codeforces

题目描述：一个n*m的网格，每个单元格上都有箭头和步长，根据箭头和步长可以移动到下一个单元格中，可以任意选择一个单元格作为开始，如果决定勾选的单元格不存在，或者如果该单元格已经勾选，则游戏结束。问是否可以选择一个单元格开始，勾选完整个网格。

题目思路：

遍历整个图，记录每个点的入度和出度。

在联通的情况下，有两种符合要求的情况：

1.整张图连成了一个环，每个点的出度和入度都是1.

2.整张图连成了一个链，第一个节点的出度会比入度大1，最后一个节点的出度会比入度小1。

其他输出No即可。

代码实现：

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5+10,M=1e6+10;
int t,n,m;
string a[N];
int f[N*10],in[N*10],out[N*10];

void init()
{
	for(int i=1;i<=n*m;i++)
	{
		f[i]=i;
		in[i]=0;
		out[i]=0;
	}
}

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

void addup(int x,int y)
{
	int xx=find(x),yy=find(y);
	if(xx!=yy)f[xx]=yy;
	
}

int main()
{
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		init();
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
			a[i]=" "+a[i];
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				int x;
				cin>>x;
				if(a[i][j]=='u'&&i-x>0)
				{
					in[((i-1)-x)*m+j]++;
					out[(i-1)*m+j]++;
					addup(((i-1)-x)*m+j,(i-1)*m+j);
				}
				if(a[i][j]=='d'&&i+x<=n)
				{
					in[((i-1)+x)*m+j]++;
					out[(i-1)*m+j]++;
					addup(((i-1)+x)*m+j,(i-1)*m+j);
				}
				if(a[i][j]=='l'&&j-x>0)
				{
					in[(i-1)*m+j-x]++;
					out[(i-1)*m+j]++;
					addup((i-1)*m+j-x,(i-1)*m+j);
				}
				if(a[i][j]=='r'&&j+x<=m)
				{
					in[(i-1)*m+j+x]++;
					out[(i-1)*m+j]++;
					addup((i-1)*m+j+x,(i-1)*m+j);
				}
			}
		}
		int f1=0,f2=0,cnt=0,pd=0;
		for(int i=1;i<=n*m;i++)
		{
			if(find(i)==i)pd++;
			if(pd>1)break;
			if(in[i]!=out[i])
			{
				cnt++;
				if(in[i]==out[i]+1) f1++;
				else if(in[i]+1==out[i]) f2++;
			}
		}
		if((pd==1&&cnt==0)||(pd==1&&cnt==2&&f1==1&&f2==1))cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}

参考博客：ZOJ-4127 Grid with Arrows-CSDN博客

END---