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%

做法：

对每个集合排序后去重，再用双指针依次比对即可。

代码：

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
using namespace std;

const int N = 10010;

vector<int> allNums[60];

int n, m, k, x;

int getSameCnt(vector<int> & a, vector<int> & b){
    int i = 0, j = 0, res = 0;
    while(i < a.size() && j < b.size()){
        if(a[i] < b[j]) i ++;
        else if(a[i] > b[j]) j ++;
        else i ++, res ++;
    }
    return res;
}

int main(){
    cin >> n;
    for(int i = 0; i < n; i ++){
        cin >> m;
        for(int j = 0; j < m; j ++){
            scanf("%d", &x);
            allNums[i].push_back(x);
        }
        sort(allNums[i].begin(), allNums[i].end());
        allNums[i].erase(unique(allNums[i].begin(), allNums[i].end()), allNums[i].end());
    }
    cin >> k;
    int a, b;
    while(k --){
        scanf("%d %d", &a, &b);
        int cnt = getSameCnt(allNums[a - 1], allNums[b - 1]);
        int cnt2 = (int)allNums[a - 1].size() + (int)allNums[b - 1].size() - cnt;
        printf("%.2f%%\n", cnt * 1.0 / cnt2 * 100);
    }
}