PTA L2-005 集合相似度 (set)

最新推荐文章于 2023-04-20 23:58:18 发布

xcatf

最新推荐文章于 2023-04-20 23:58:18 发布

阅读量580

点赞数 1

CC 4.0 BY-SA版权

分类专栏： OJ基础题文章标签： PTA

本文链接：https://blog.youkuaiyun.com/qq1013459920/article/details/85123543

OJ基础题专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一个算法问题，即计算两个整数集合之间的相似度。通过使用C++ STL中的set，文章提供了一种高效的方法来计算两个集合共有元素的比例，并给出了完整的代码实现。

L2-005 集合相似度（25 分）

给定两个整数集合，它们的相似度定义为：Nc/Nt×100%。其中Nc是两个集合都有的不相等整数的个数，Nt是两个集合一共有的不相等整数的个数。你的任务就是计算任意一对给定集合的相似度。

输入格式：

输入第一行给出一个正整数N（≤50），是集合的个数。随后N行，每行对应一个集合。每个集合首先给出一个正整数M（≤104），是集合中元素的个数；然后跟M个[0,109]区间内的整数。

之后一行给出一个正整数K（≤2000），随后K行，每行对应一对需要计算相似度的集合的编号（集合从1到N编号）。数字间以空格分隔。

输出格式：

对每一对需要计算的集合，在一行中输出它们的相似度，为保留小数点后2位的百分比数字。

输入样例：

3
3 99 87 101
4 87 101 5 87
7 99 101 18 5 135 18 99
2
1 2
1 3

输出样例：

50.00%
33.33%

不用STL的话写这个题会比较麻烦，需要自己写一颗平衡树或者进行数组排序后二分查找

可以使用STL中的set(基于红黑树实现)，实现去查以及集合中二分查找

AC Code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
static const int N=1e5+5;
set<int>s[55];
int main(){
    int n,m,x,y;
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&m);
        for(int j=0;j<m;++j){
            scanf("%d",&x);
            s[i].insert(x);
        }
    }
    scanf("%d",&m);
    for(int i=0;i<m;++i){
        scanf("%d%d",&x,&y);
        int b=s[x].size()+s[y].size(),a=0;
        for(auto it=s[x].begin();it!=s[x].end();++it){
            if(s[y].count(*it)) ++a;    //判断y集合中是否含有x集合中该元素
        }
        printf("%.2f%%\n",1.0*a/(b-a)*100);
    }
    return 0;
}