1402. 星空之夜 （DFS+连通块相似判断）AcWing寒假每日一题 提高组（1）

题目链接

题目：

夜空深处，闪亮的星星以星群的形式出现在人们眼中，形态万千。

一个星群是指一组非空的在水平，垂直或对角线方向相邻的星星的集合。

一个星群不能是一个更大星群的一部分。

星群可能是相似的。

如果两个星群的形状、包含星星的数目相同，那么无论它们的朝向如何，都认为它们是相似的。

通常星群可能有 8 种朝向，如下图所示：

现在，我们用一个二维 01 矩阵来表示夜空，如果一个位置上的数字是 1，那么说明这个位置上有一个星星，否则这个位置上的数字应该是 0。

给定一个夜空二维矩阵，请你将其中的所有星群用小写字母进行标记，标记时相似星群用同一字母，不相似星群用不同字母。

标注星群就是指将星群中所有的 1 替换为小写字母。

输入格式
第一行包含一个整数 W，表示矩阵宽度。

第二行包含一个整数 H，表示矩阵高度。

接下来 H 行，每行包含一个长度为 W 的 01 序列，用来描述整个夜空矩阵。

输出格式
输出标记完所有星群后的二维矩阵。

用小写字母标记星群的方法很多，我们将整个输出读取为一个字符串，能够使得这个字符串字典序最小的标记方式，就是我们想要的标记方式。

输出这个标记方式标出的最终二维矩阵。

数据范围
0≤W,H≤100,
0≤ 星群数量 ≤500,
0≤ 不相似星群数量 ≤26,
1≤ 星群中星星的数量 ≤160
输入样例：

23
15
10001000000000010000000
01111100011111000101101
01000000010001000111111
00000000010101000101111
00000111010001000000000
00001001011111000000000
10000001000000000000000
00101000000111110010000
00001000000100010011111
00000001110101010100010
00000100110100010000000
00010001110111110000000
00100001110000000100000
00001000100001000100101
00000001110001000111000

输出样例：

a000a0000000000b0000000
0aaaaa000ccccc000d0dd0d
0a0000000c000c000dddddd
000000000c0b0c000d0dddd
00000eee0c000c000000000
0000e00e0ccccc000000000
b000000e000000000000000
00b0f000000ccccc00a0000
0000f000000c000c00aaaaa
0000000ddd0c0b0c0a000a0
00000b00dd0c000c0000000
000g000ddd0ccccc0000000
00g0000ddd0000000e00000
0000b000d0000f000e00e0b
0000000ddd000f000eee000

样例解释
样例对应的星空图如下：

答案对应的标记后星空图如下：

思路：

• 这题是很明显的求连通性的题目，但是在我们找到一个连通块之后，如何存储此连通块的8种朝向（即判断是否同构或者判断相似）和该连通块对应的字符就是一个难点。
• 这里我们可以用到一个映射来对解决第一个判断同构的问题，我们先用dfs将一个连通块中的所有点存入一个vector，然后再计算得出在这个连通块中点和点的距离之和作为标识一个连通块的key值。
• 处理完同构之后，我们只需开一个map来存储每个连通块的字符即可，通过比较我们上一步计算得出的距离之和来判断这个连通块是否存入过我们开设的map，如果有，则直接返回相应字符，如果无，则将这个连通块存入map中。

代码：

#include<iostream>
#include<vector>
#include<stdio.h>
#include<map>
#include<cmath>
using namespace std;
#define x first
#define y second
int h,w;
int cnt;
char g[105][105];
int dx[8]={0,0,1,-1,1,-1,1,-1};
int dy[8]={1,-1,1,-1,-1,1,0,0};
typedef pair<int,int> PII;
vector<PII> temp;
map<double,char> mp;
void dfs(int x,int y)
{
    g[x][y]='0';
    temp.push_back(make_pair(x,y));
    for(int i=0;i<8;i++)
    {
        int xx=x+dx[i];
        int yy=y+dy[i];
        if(xx>=0&&yy>=0&&xx<h&&yy<w&&g[xx][yy]=='1')
        {
            dfs(xx,yy);
        }
    }
}
double get_dist(PII a,PII b)
{
    double deltx=a.x-b.x,delty=a.y-b.y;
    return sqrt(deltx*deltx+delty*delty);
}
double get_hash()
{
    double res=0;
    for(int i=0;i<temp.size();i++)
    {
        for(int j=0;j<temp.size();j++)
        {
            res+=get_dist(temp[i],temp[j]);
        }
    }
    return res;
}
char get_ch(double sum)
{
    for(auto s:mp)
    {
        if(abs(s.x-sum)<1e-6)
        {
            return s.y;
        }
    }
    mp[sum]=mp.size()+'a';
    return mp[sum];
}
void change(char ch)
{
    for(int i=0;i<temp.size();i++)
    {
        int xx=temp[i].x,yy=temp[i].y;
        g[xx][yy]=ch;
    }
}
int main()
{

    cin>>w>>h;
    for(int i=0;i<h;i++)
    {
        for(int j=0;j<w;j++)
        {
            cin>>g[i][j];
        }
    }
     for(int i=0;i<h;i++)
    {
        for(int j=0;j<w;j++)
        {
            if(g[i][j]=='1')
            {
                dfs(i,j);
                double cur_hash=get_hash();
                char ch=get_ch(cur_hash);
                change(ch);
                temp.clear();
            }
        }
    }
     for(int i=0;i<h;i++)
    {
        for(int j=0;j<w;j++)
        {
            cout<<g[i][j];
        }
        cout<<endl;
    }
}