https://www.patest.cn/contests/pat-a-practise/1063
求交集时,不要两重循环从头找到尾,因为这样会超时On2。
可以将两个集合分别排好序再查找,时间复杂度是On。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
constexpr int N=51;
constexpr int K=2000;
struct query {
int x;
int y;
};
struct query qy[K];
float calPer (vector<int> v1, vector<int> v2);
int cmp (int a, int b) {
return a<b;
}
int main () {
int n, m, k;
vector<int> va[N];
int i=1;
int j;
scanf ("%d", &n);
while (i <= n) {
scanf ("%d", &m);
j=0;
int temp;
while (j < m) {
scanf ("%d", &temp);
if (find (va[i].begin(), va[i].end(), temp) == va[i].end()) //去重
va[i].push_back(temp);
j++;
}
sort (va[i].begin(), va[i].end(), cmp);
i++;
}
scanf ("%d", &k);
i=0;
int tx, ty;
while (i < k) {
scanf ("%d %d", &tx, &ty);
qy[i].x=tx;
qy[i].y=ty;
i++;
}
i=0;
while (i < k) {
tx=qy[i].x;
ty=qy[i].y;
float per=calPer (va[tx], va[ty]);
printf ("%.1f%%\n", per);
i++;
}
return 0;
}
float calPer (vector<int> v1, vector<int> v2) {
int n1=v1.size();
int n2=v2.size();
int cnt=0;
auto iter2=v2.begin();
for (auto iter1=v1.begin(); iter1!=v1.end() && iter2!=v2.end(); iter1++) { //这里不要两个循环遍历,这样会超时, 先将va[i]排序, 这样在这里就不耗时了。。
while (iter2 != v2.end()) {
if (*iter1 == *iter2) {
iter2++;
cnt++;
} else if (*iter1 < *iter2) {
break;
} else {
iter2++;
}
}
}
float p=(float)cnt/(n1+n2-cnt);
p=p*100;
return p;
}