1. 程式人生 > >PTAL2-005 集合相似度解題報告---set查詢

PTAL2-005 集合相似度解題報告---set查詢

                                   L2-005 集合相似度 (25 分)

給定兩個整數集合,它們的相似度定義為:N​c​​/N​t​​×100%。其中N​c​​是兩個集合都有的不相等整數的個數,N​t​​是兩個集合一共有的不相等整數的個數。你的任務就是計算任意一對給定集合的相似度。

輸入格式:

輸入第一行給出一個正整數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%

不用STL的話這個題拿滿分就比較麻煩了,要自己寫一顆二叉樹或或者進行陣列排序後二分查詢

STL中的set基於二叉樹實現,自帶搜尋find函式,解決最後一個測試點超時的情況

 AC Code:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<string>
#include<cctype>
#include<map>
#include<vector>
#include<string>
#include<queue>
#include<stack>
#include<set>
#define INF 0x3f3f3f3f
using namespace std;
static const int MAX_N = 1e5 + 5;
typedef long long ll;
set<int> S[55];
int main(){
	int n;
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
		int m;
		scanf("%d", &m);
		for (int j = 0; j < m; j++) {
			int v1;
			scanf("%d", &v1);
			S[i].insert(v1);	//加入集合(自動去重)
		}
	}
	int q;
	scanf("%d", &q);
	while (q--) {
		int c1, c2;
		scanf("%d%d", &c1, &c2);
		c1--; c2--;
		int nc = S[c1].size(), nt = S[c1].size() + S[c2].size();
		for (set<int>::iterator it = S[c1].begin(); it != S[c1].end(); it++) {
			if (S[c2].find(*it) != S[c2].end()) nt--;	//find二叉樹搜尋
			else nc--;
		}
		printf("%.2f%%\n", 1.0 * nc / nt * 100);
	}
	return 0;
}