PAT A1063 Set Similarity

题目链接

最后一个测试点超时

原因

求两个集合不同的个数时，不要新建一个集合，然后都插入进去！

代码

#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
const int N=55;
const int M=1e4+10;

set<int> sets[N];
bool flag[M]={false};
int main(){
    int n,m,x,k,s1,s2;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&m);
        for(int j=0;j<m;j++){
            scanf("%d",&x);
            sets[i].insert(x);
        }
    }
    scanf("%d",&k);
    for(int i=0;i<k;i++){
        scanf("%d%d",&s1,&s2);
        //遍历s1集合，判断在s2中是否存在，求相同数的个数
        double nc=0,nt=0,ans;
        set<int> set_sum;
        for(auto it=sets[s1].begin();it!=sets[s1].end();it++){
            if(sets[s2].find((*it))!=sets[s2].end()) nc++;
        }
     	//使用这个导致超时！！！
        set_sum.insert(sets[s1].begin(),sets[s1].end());
        set_sum.insert(sets[s2].begin(),sets[s2].end());
        nt=set_sum.size();
        ans=nc/nt*100;
        printf("%.1f%\n",ans);
    }
    return 0;
}

AC解法

思路

求解两个集合中不同的集合，就是一个集合的总数加上另一个集合中与该集合不同的个数！！！

代码

#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
const int N=55;
const int M=1e4+10;

set<int> sets[N];
bool flag[M]={false};
int main(){
    int n,m,x,k,s1,s2;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&m);
        for(int j=0;j<m;j++){
            scanf("%d",&x);
            sets[i].insert(x);
        }
    }
    scanf("%d",&k);
    for(int i=0;i<k;i++){
        scanf("%d%d",&s1,&s2);
        //遍历s1集合，判断在s2中是否存在，求相同数的个数
        double nc=0,nt=sets[s2].size(),ans;
        set<int> set_sum;
        for(auto it=sets[s1].begin();it!=sets[s1].end();it++){
            if(sets[s2].find((*it))!=sets[s2].end()) nc++;//不等于end说明s2中存在
            else nt++;
        }
        ans=nc/nt*100;
        printf("%.1f%\n",ans);
    }
    return 0;
}