起初尝试使用map+set嵌套解决该题(猜想过可能会有一组数据超时,果不其然最后一个测试点超时)
结论:在PAT中考察容器的使用,切记:不可太过省事,尽量少使用容器解决此类题目,因为一般都有时间限制,完完全全使用容器会容易导致超时。
如下代码就是一个很好的例子
(但是该代码有时能过测试点4,读者可尝试多跑几次)
#include<iostream>
#include<set>
#include<map>
using namespace std;
map<int, set<int>> setMP;
double similarity(int set1, int set2) {
auto setA = setMP[set1];
auto setB = setMP[set2];
int set1Size = setA.size();
int set2Size = setB.size();
for(auto it = setB.begin(); it != setB.end(); it++) {
setA.insert(*it); // 该操作比较耗时,插入操作时间复杂度为logN
}
int total = setA.size();
return 1.0 * (set1Size + set2Size - total) / total * 100;
}
int main() {
int n, m, k, setItem;
scanf("%d", &n);
for(int i = 1; i <= n; i++) {
scanf("%d", &m);
for(int j = 0; j < m; j++) {
scanf("%d", &setItem);
setMP[i].insert(setItem);
}
}
scanf("%d", &k);
int set1, set2;
for(int i=0; i<k; i++) {
scanf("%d%d", &set1, &set2);
printf("%.1f%\n", similarity(set1, set2));
}
return 0;
}
修改代码
去掉了map和set的嵌套 、测试点4还是超时,但是有时能通过
将下述代码中的set改为unordered_set也无效,测试点4依然超时
#include<iostream>
#include<set>
#include<map>
using namespace std;
set<int> setMP[55];
double similarity(int set1, int set2) {
auto setA = setMP[set1];
auto setB = setMP[set2];
int set1Size = setA.size();
int set2Size = setB.size();
for(auto it = setB.begin(); it != setB.end(); it++) {
setA.insert(*it);
}
int total = setA.size();
return 1.0 * (set1Size + set2Size - total) / total * 100;
}
int main() {
int n, m, k, setItem;
scanf("%d", &n);
for(int i = 1; i <= n; i++) {
scanf("%d", &m);
for(int j = 0; j < m; j++) {
scanf("%d", &setItem);
setMP[i].insert(setItem);
}
}
scanf("%d", &k);
int set1, set2;
for(int i=0; i<k; i++) {
scanf("%d%d", &set1, &set2);
printf("%.1f%\n", similarity(set1, set2));
}
return 0;
}
最优解(和上述代码比较,只是将插入操作改为了查找操作)
查找资料发现:插入操作和查找操作的时间复杂度都为logN,
但通过这题可以看出,查找操作耗时较少
#include<iostream>
#include<set>
#include<map>
using namespace std;
set<int> setMP[55];
double similarity(int set1, int set2) {
auto setA = setMP[set1];
auto setB = setMP[set2];
int set1Size = setA.size(), same=0;
for(auto it = setB.begin(); it != setB.end(); it++) {
if(setA.find(*it) != setA.end()) same++;
else set1Size++;
}
return 1.0 * same / set1Size * 100;
}
int main() {
int n, m, k, setItem;
scanf("%d", &n);
for(int i = 1; i <= n; i++) {
scanf("%d", &m);
for(int j = 0; j < m; j++) {
scanf("%d", &setItem);
setMP[i].insert(setItem);
}
}
scanf("%d", &k);
int set1, set2;
for(int i=0; i<k; i++) {
scanf("%d%d", &set1, &set2);
printf("%.1f%\n", similarity(set1, set2));
}
return 0;
}