团体程序设计天梯赛GPLT——L2-005 集合相似度

点击转到PTA题目

输入格式：

输入第一行给出一个正整数N（≤50），是集合的个数。随后N行，每行对应一个集合。每个集合首先给出一个正整数M（≤10 ^ 4），是集合中元素的个数；然后跟M个[0,10 ^ 9]区间内的整数。之后一行给出一个正整数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%
(结尾无空行)

题目分析
看到去重就想到利用set了，但是缺采取了一种最笨的办法，就是没有在输入的同时做插入，而是在每次查询的时候才开始插入set中，这就导致了数据量大的时候程序浪费在插入操作中的时间消耗是巨大的，因此在最后一个测试点超了时间，只拿了21分。

#include<bits/stdc++.h>
using namespace std;
const int num=10005;
int N,M,loop[num];
set<int> unequal;
set<int> equal_1;
set<int> equal_2;
long arr[num][num];
void calculate(int x,int y)
{
	for(int i=0;i<loop[x];i++) 
	{
		unequal.insert(arr[x][i]);
		equal_1.insert(arr[x][i]);
	}
	for(int i=0;i<loop[y];i++) 
	{
		unequal.insert(arr[y][i]);
		equal_2.insert(arr[y][i]);
 }
	int N2=unequal.size();
	int N1=0;	
	for(set<int>::iterator i=equal_1.begin();i!=equal_1.end();i++)
	{ 
		for(set<int>::iterator j=equal_2.begin();j!=equal_2.end();j++)
		{
			if(*i==*j)
			{
				N1++;
				break;
			}		     	
		}	
	}
	float result=(float)N1/N2;
	result*=100;
	cout << setiosflags(ios::fixed) << setprecision(2) << result << "%" << endl;
}
int main()
{
	cin >> N;
	for(int i=0;i<N;i++)
	{
		cin >> loop[i];
		for(int j=0;j<loop[i];j++)
		{
			cin >> arr[i][j];
		}		
	}
	int test,judge[N][N];
	cin >> test;
	for(int i=0;i<test;i++)
	{
		int x,y;
		cin >> x >> y;
		unequal.clear();
        equal_1.clear();
        equal_2.clear();
		calculate(x-1,y-1);		
	}
	return 0;
}

而后即使利用二分等操作去尽可能的减少时间复杂度，但还是抵消不了插入操作的影响，最终转变思路，采取了边输入边插入的方式，更关键的是找到了len1+len2-M这个关键关系，就不用再设立一个集合来维护Nt了，大大节省了时间。
关键是得利用好set的特性，以及一些成员函数，这些都是我们在准备竞赛时需要重点掌握的，而且还要知道没有如果不用set怎么去重的方法，这些都很重要。

#include<bits/stdc++.h>
using namespace std;
int N;
set<int> equal_set[55];
set<int> abc;
int main()
{
	cin >> N;
	for(int i=1;i<=N;i++)
	{
		int loop,x;
		cin >> loop;
		for(int j=0;j<loop;j++)
		{
			cin >> x;
			equal_set[i].insert(x);
		}		
	}
	int test;
	cin >> test;
	for(int i=0;i<test;i++)
	{
		int x,y;
		cin >> x >> y;
		int len1=equal_set[x].size();
		int len2=equal_set[y].size();
		int M=0;
        for(set<int>::iterator it=equal_set[x].begin();it!=equal_set[x].end();it++)
	   {
		   if(equal_set[y].find(*it)!=equal_set[y].end())
		  {
		     M++;	
		  }	
	   }
	float result=(float)M/(len1+len2-M);
	result*=100;
	cout << setiosflags(ios::fixed) << setprecision(2) << result << "%" << endl;			
	}
	return 0;
}