PTA | 集合相似度

目录

题目：

输入格式：

输出格式：

输入样例：

输出样例：

代码：

无注释版：

有注释版：

---

题目：

给定两个整数集合，它们的相似度定义为：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%

代码长度限制16 KB，时间限制500 ms，内存限制64 MB，栈限制8192 KB

代码：

C++ (clang++）

无注释版：

#include<bits/stdc++.h>
using namespace std;
#define int long long
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin>>n;
	vector<set<int>> a(n);
	for(int i=0;i<n;i++){
		int m;
		cin>>m;
		for(int j=0;j<m;j++){
			int x;
			cin>>x;
			a[i].insert(x);
		}
	} 
	int k;
	cin>>k;
	for(int i=0;i<k;i++){
		int x,y;
		cin>>x>>y;
		x--;
		y--;
		set<int> p;
		set_intersection(a[x].begin(),a[x].end(),a[y].begin(),a[y].end(),
                         inserter(p,p.begin()));
        set<int> q;
        set_union(a[x].begin(),a[x].end(),a[y].begin(),a[y].end(),
                  inserter(q,q.begin()));
		double s=(p.size()*1.0/q.size())*100;
		cout<<fixed<<setprecision(2)<<s<<"%\n";
	}
}

有注释版：

#include <bits/stdc++.h>  // 包含所有标准库的头文件，加快编写代码速度
using namespace std;

#define int long long  // 定义 `int` 为 `long long`，防止整数溢出

// 主函数
signed main() {
    ios::sync_with_stdio(0); // 关闭 `cin` 和 `cout` 的同步，提高输入输出效率
    cin.tie(0); // 解绑 `cin` 和 `cout`，进一步提高效率
    cout.tie(0);

    int n; // `n` 表示集合的个数
    cin >> n;

    vector<set<int>> a(n); // 用 `vector` 存储 `set`，每个 `set` 代表一个集合

    // 读取 `n` 个集合
    for (int i = 0; i < n; i++) {
        int m; // `m` 表示当前集合的元素个数
        cin >> m;
        for (int j = 0; j < m; j++) {
            int x; 
            cin >> x; // 读取集合元素
            a[i].insert(x); // 使用 `set` 自动去重，确保集合中的元素唯一
        }
    }

    int k; // `k` 表示需要计算相似度的集合对的个数
    cin >> k;

    // 处理 `k` 组查询
    for (int i = 0; i < k; i++) {
        int x, y;
        cin >> x >> y; // 读取需要计算相似度的集合编号
        x--; // 因为输入是从 1 开始编号，而 `vector` 从 0 开始，需要转换
        y--;

        set<int> p; // `p` 存储两个集合的交集（共有的不相等整数）
        set_intersection(a[x].begin(), a[x].end(), a[y].begin(), a[y].end(),
                         inserter(p, p.begin()));

        set<int> q; // `q` 存储两个集合的并集（所有不相等的整数）
        set_union(a[x].begin(), a[x].end(), a[y].begin(), a[y].end(),
                  inserter(q, q.begin()));

        double s = (p.size() * 1.0 / q.size()) * 100; // 计算相似度并转换为百分比
        cout << fixed << setprecision(2) << s << "%\n"; // 保留两位小数，并输出百分比格式
    }
}