PTA L2-005 集合相似度

PTA L2-005 集合相似度

---

题目

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

输入

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

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

输出

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

---

思路

创建集合数组，然后用set自身的find来找相同的元素

---

最终代码

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
#include<iomanip>
using namespace std;

int main(){
    vector<set<int>> A;
    set<int> Ax;
    int N,M,A0,A1,K,x;
    double Nc;
    cin >> N;
    for(int i=0;i<N;i++){
        cin >> M;
        for(int j=0;j<M;j++){
            cin >> x;
            Ax.insert(x);
        }
        A.push_back(Ax);
        Ax.clear();
    }
    cin >> K;
    for(int i=0;i<K;i++){
        cin >> A0 >> A1;
        A0--;A1--;
        Nc=0;
        for(set<int>::iterator it = A[A0].begin();it!=A[A0].end();it++){
            if(A[A1].find(*it)!=A[A1].end()){
                Nc+=1;
            }
        }
        cout << fixed << setprecision(2) << Nc*100/(A[A0].size()+A[A1].size()-Nc) << '%' << endl;
    }
    return 0;
}

---

遇过的问题

• Ax没有clear掉，导致前面的集合影响后面的集合

• 运行超时

---

总结

1. 临时的set，vector等，用完一定要clear，否则会影响后面
2. set本身的find复杂度比algorithm里vector的find复杂度低，所以直接用set.find(*it)比用find(vec.begin(),vec.end(),*it)要快