题解：CF1600J Robot Factory

Robot Factory

思路

这题其实就是另类一点的连通块问题。每一步都要通过位运算判断可否前行。这题关键的是如何判断能否前向，其实通过与运算就可以解决这个问题。

代码

#include<bits/stdc++.h>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#define ll long long
using namespace std;
const int N=1e5+10;
const int M=2023;
const int inf=0x3f3f3f3f;
int n,m;
int a[1145][1145];
int vis[1145][1145];
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};
int dk[]={4,1,2,8};//判断是否可行
int h[N],num;
int ans;
bool cmp(int x,int y)
{
	return x>y;
}
void dfs(int x,int y)
{
	ans++;
	for(int i=0;i<4;i++)
	{
		int nx=x+dx[i];
		int ny=y+dy[i];
		int k=a[x][y]&dk[i];//0才可以通过
		if(nx>=1 and nx<=n and ny>=1 and ny<=m and vis[nx][ny]==0 and !k)
		{
			vis[nx][ny]=1;
			dfs(nx,ny);
		}
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			cin>>a[i][j];
		}
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(vis[i][j]==0)
			{
				ans=0;
				vis[i][j]=1;
				dfs(i,j);
				num++;
				h[num]=ans;
			}
		}
	}
	sort(h+1,h+num+1,cmp);
	for(int i=1;i<=num;i++)
	{
		cout<<h[i]<<" ";
	}
	return 0;
}