相似图片分类 [华为]【并查集】

题目描述:

小明想要处理一批图片,将相似的图片分类,他首先对图片的特征采样,得到图片之间的相似度,然后

按照以下规则判断图片是否可以归为一类:

1)相似度>0表示两张图片相似;

2)如果A和B相似,B和C相似,但A和C不相似,那么认为A和C间接相似,可以把ABC归为一类,但不计算AC的相似度;

3)如果A和所有其他图片都不相似,则A自己归为一类,相似度为0。

给定一个大小为N×N的矩阵M存储任意两张图片的相似度,M[i][j]即为第i个图片和第j个图片的相似度,请按照"从大到小"的顺序返回每个相似类

中所有图片的相似度之和。

输入描述:

第一行为一个整数N,代表矩阵M中有N个图片。下面跟着N行,每行有N列数据,空格分隔(为了显示整齐,空格可能为多个),代表N个图片之间的相似度。

约束:

  1. 0 < N <= 900
  2. 0 <= M[i][j] <= 100,输入保证 M[i][i]=0,M[i][j]=M[j][i]
  3. 输入的矩阵中分隔符为1个或连续多个空格

输出描述:

每个相似类的相似度之和。格式为:一行数字。分隔符为1个空格

样例1

输入

5

0 0 50 0 0

0 0 0 25 0

50 0 0 0 15

0 25 0 0 0

0 0 15 0 0

输出

62 25

样例2

输入

5

0 5 42 0 0

5 0 0 91 0

42 0 0 0 15

0 91 0 0 0

0 0 15 0 0

输出

153

样例3

输入

7

0 1 3 0 0 0 0

1 0 2 0 0 0 0

3 2 0 0 0 0 0

0 0 0 0 3 0 9

0 0 0 3 0 5 0

0 0 0 0 5 0 7

0 0 0 9 0 7 0

输出

24 6

AC代码:

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;

int n, m;
int k[N][N];
int p[N];
map<int, int> p2;
map<int, bool> p3;
vector<int> res;

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

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i ++) p[i] = i;
    for(int i = 1; i <= n; i ++)
    {
        for(int j = 1; j <= n; j ++)
        {
            cin >> k[i][j];
            if(k[i][j] > 0) p[find(i)] = find(j);
        }
    }
    for(int i = 1; i <= n; i ++)
    {
        for(int j = 1; j <= n; j ++)
        {
            if(find(i) == find(j) && k[i][j] > 0) p2[find(i)] += k[i][j];
        }
    }
    for(int i = 1; i <= n; i ++)
    {
        if(p3[find(i)] == false)
        {
            res.push_back(p2[find(i)] / 2);
            p3[find(i)] = true;
        }
    }
    sort(res.begin(), res.end());
    reverse(res.begin(), res.end());
    for(int i = 0; i < res.size(); i ++)
    {
        if(i != res.size() - 1) cout << res[i] << " ";
        else cout <<  res[i];
    }
    return 0;
}
相关推荐
何其有幸.5 小时前
实验3-3 比较大小(PTA|C语言)
c语言·数据结构·算法
东阳马生架构6 小时前
Sentinel源码—8.限流算法和设计模式总结二
算法·设计模式·sentinel
老饼讲解-BP神经网络7 小时前
一篇入门之-评分卡变量分箱(卡方分箱、决策树分箱、KS分箱等)实操例子
算法·决策树·机器学习
何其有幸.7 小时前
实验6-3 使用函数求特殊a串数列和(PTA|C语言)
c语言·数据结构·算法
不会计算机的捞地7 小时前
【数据结构入门训练DAY-24】美国大选
数据结构·算法
明月看潮生7 小时前
青少年编程与数学 02-018 C++数据结构与算法 11课题、分治
c++·算法·青少年编程·编程与数学
Echo``8 小时前
2:QT联合HALCON编程—图像显示放大缩小
开发语言·c++·图像处理·qt·算法
.似水8 小时前
2025.4.22_C_可变参数列表
java·c语言·算法
Felven8 小时前
A. Ideal Generator
java·数据结构·算法
MoonBit月兔9 小时前
双周报Vol.70: 运算符重载语义变化、String API 改动、IDE Markdown 格式支持优化...多项更新升级!
ide·算法·哈希算法