L2-005 集合相似度

法一(暴力超时21分)
纯暴力,最后一个测试点超时。

#include <bits/stdc++.h>
using namespace std;
vector<set<int>> dataset;
int main(){
	int n;
	cin>>n;
	dataset.resize(n+1);
    for(int i=1;i<=n;i++){
    	int count,data;
    	cin>>count;
    	for(int j=0;j<count;j++){
    		cin>>data;
    		dataset[i].insert(data);
		}
	}
	int m;
	cin>>m;
	set<int>::iterator it;
	for(int i=0;i<m;i++){
		int g1,g2;
		cin>>g1>>g2;
		set<int> st;
		for(it=dataset[g1].begin();it!=dataset[g1].end();++it){
			st.insert(*it);
		}
		for(it=dataset[g2].begin();it!=dataset[g2].end();++it){
			st.insert(*it);
		}
		int Nt=st.size();
		int Nc=dataset[g1].size()+dataset[g2].size()-Nt;
		//cout<<Nt<<" "<<Nc<<endl;
		cout<<fixed<<setprecision(2)<<(1.0*Nc/Nt*10000)/100 << "%"<<endl;
	}
	return 0;
}

法二stl(25分)

#include <bits/stdc++.h>
using namespace std;
vector<set<int>> dataset;
int main(){
	int n;
	cin>>n;
	dataset.resize(n+1);
    for(int i=1;i<=n;i++){
    	int count,data;
    	cin>>count;
    	for(int j=0;j<count;j++){
    		cin>>data;
    		dataset[i].insert(data);
		}
	}
	int m;
	cin>>m;
	set<int>::iterator it;
	for(int i=0;i<m;i++){
		int g1,g2;
		cin>>g1>>g2;
		int cm=0;
		for(it=dataset[g1].begin();it!=dataset[g1].end();++it){
			if(dataset[g2].find(*it)!=dataset[g2].end()){
				cm++;
			}
		}
		int Nt=dataset[g1].size()+dataset[g2].size()-cm;
		int Nc=cm;
		cout<<fixed<<setprecision(2)<<(1.0*Nc/Nt*10000)/100 << "%"<<endl;
	}
	return 0;
}
posted @ 2024-03-14 16:59  YuKiCheng  阅读(13)  评论(0编辑  收藏  举报