Bubble Cup 14 - Finals Online Mirror (Unrated, ICPC Rules, Teams Preferred, Div. 2) J. Robot Factory

传送门
请添加图片描述

题意:

给你一个矩阵,矩阵中的每一个数的二进制串可以表示出该位置周围的围墙信息,让你求出所有的房间大小。

思路:

建图,搜索,一气呵成。

#include<iostream>
#include<string>
#include<map>
#include<bits/stdc++.h>
#include<iomanip>
#include<algorithm>
#define ll long long
using namespace std;
int g[4050][4050];
int bit[10];
void get_bit(int a)
{
    
    
	int cnt = 0;
	while(a)
	{
    
    
		bit[++cnt] = a%2;
		a = a>>1; 
	}
}
int h[4] = {
    
    2,0,-2,0},l[4] = {
    
    0,2,0,-2};
int n,m;
int ans[1000010];
int maxx = 0;
int exchange(int x)
{
    
    
	if(!x)return x;
	int t = x/abs(x);
	return t;
}
void dfs(int i, int j)
{
    
    
	for(int k = 0; k < 4; k++)
	{
    
    
		if(!g[i+h[k]][j+l[k]] && !g[i+exchange(h[k])][j+exchange(l[k])] && i+h[k] <= (n-1)*2+4 && i+h[k] >= 4 && j+l[k] <= (m-1)*2+4 && j+l[k] >= 4)
		{
    
    
			maxx++;
			g[i+h[k]][j+l[k]] = 1;
			dfs(i+h[k], j+l[k]);
		}
	}
}
int main()
{
    
    

    cin>>n>>m;
    for(int i = 4; i <= (n-1)*2+4; i+=2)
	{
    
    
		for(int j = 4; j <= (m-1)*2+4; j+=2)
		{
    
    
			int a;
			cin>>a;
			get_bit(a);
			for(int k = 4; k >= 1; k--)
			{
    
    
				if(k == 4 && bit[k])g[i-1][j] = 1;
				else if(k == 3 && bit[k])g[i][j+1] = 1;
				else if(k == 2 && bit[k])g[i+1][j] = 1;
				else if(k == 1 && bit[k])g[i][j-1] = 1;
			}
			memset(bit,0,sizeof(bit));
		}
	}

	int num = 0;
	for(int i = 4; i <= (n-1)*2+4; i+=2)
	{
    
    
		for(int j = 4; j <= (m-1)*2+4; j+=2)
		{
    
    
			maxx = 1;
			if(!g[i][j])
			{
    
    
				g[i][j] = 1;
				dfs(i,j);
				ans[++num] = maxx;

			}
		}
	}
	sort(ans+1,ans+1+num);
	for(int i = num; i >= 1; i--)cout<<ans[i]<<" ";
}


猜你喜欢

转载自blog.csdn.net/p15008340649/article/details/120895534