7-27 秀恩爱分得快

古人云：秀恩爱，分得快。

互联网上每天都有大量人发布大量照片，我们通过分析这些照片，可以分析人与人之间的亲密度。如果一张照片上出现了 K 个人，这些人两两间的亲密度就被定义为 1/K。任意两个人如果同时出现在若干张照片里，他们之间的亲密度就是所有这些同框照片对应的亲密度之和。下面给定一批照片，请你分析一对给定的情侣，看看他们分别有没有亲密度更高的异性朋友？

输入格式：

输入在第一行给出 2 个正整数：N（不超过1000，为总人数——简单起见，我们把所有人从 0 到 N-1 编号。为了区分性别，我们用编号前的负号表示女性）和 M（不超过1000，为照片总数）。随后 M 行，每行给出一张照片的信息，格式如下：

K P[1] ... P[K]

其中 K（≤ 500）是该照片中出现的人数，P[1] ~ P[K] 就是这些人的编号。最后一行给出一对异性情侣的编号 A 和 B。同行数字以空格分隔。题目保证每个人只有一个性别，并且不会在同一张照片里出现多次。

输出格式：

首先输出 A PA，其中 PA 是与 A 最亲密的异性。如果 PA 不唯一，则按他们编号的绝对值递增输出；然后类似地输出 B PB。但如果 A 和 B 正是彼此亲密度最高的一对，则只输出他们的编号，无论是否还有其他人并列。

输入样例 1：

10 4
4 -1 2 -3 4
4 2 -3 -5 -6
3 2 4 -5
3 -6 0 2
-3 2

输出样例 1：

-3 2
2 -5
2 -6

输入样例 2：

4 4
4 -1 2 -3 0
2 0 -3
2 2 -3
2 -1 2 
-3 2

输出样例 2：

-3 2

题解：

sex[N]标记性别，string转换成int存入unordered_set去重

找a，b所在的照片及对存在的异性记录亲密度

寻找并记录pa，pb

注意 -0 的存在，及-0从string向int转化为0

#include<bits/stdc++.h>
#include<unordered_set>
using namespace std;
const int  N = 1005;
int n, m, k;
string a, b;
int sex[N];
unordered_set<int>ph[N];
map<int, double>A, B;
int main() {
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= m; i++) {//存照片
		scanf("%d", &k);
		string t;
		for (int j = 1; j <= k; j++) {
			cin >> t;
			int tem;
			istringstream ss(t);ss >> tem;
			if (t[0] == '-')sex[abs(tem)] = 1;
			ph[i].insert(abs(tem));
		}
	}cin >> a >> b;
	int ta, tb;
	istringstream ssa(a);ssa >> ta;
	if (a[0] == '-')ta=abs(ta),sex[ta] = 1;
	istringstream ssb(b);ssb >> tb;
	if (b[0] == '-')tb=abs(tb),sex[tb] = 1;
	

	for (int i = 1; i <= m; i++) {//找a，b所在的照片及对存在的异性记录亲密度
		if (ph[i].find(ta) != ph[i].end()) {
			//cout << "a:" << a << endl;
			for (auto j : ph[i]) {
				if (sex[ta]^sex[j] ) {
					double x = ph[i].size();
					A[j] += 1.0 / x;
					/*printf("x:%lf A[", x);
					cout << j;
					printf("]:% lf\n", A[j]);*/
				}
			}
		}//printf("%d %d", A.size(), B.size());
		if (ph[i].find(tb) != ph[i].end()) {
			//cout << "b:" << b << endl;
			for (auto j : ph[i]) {
				if (sex[j]^sex[tb]) {
					double x = ph[i].size();
					B[j] += 1.0 / x;
					/*printf("x:%lf B[", x);
					cout << j;
					printf("]:% lf\n", B[j]);*/
				}
			}
		}
	}
	set<int>pa, pb;
	double am = 0, bm = 0;
	
	for (auto i : A) {//找pa，pb
		if (i.second - am > 1e-7) {
			pa.clear();
			pa.insert(i.first);
			am = i.second;
		}
		else if (abs(i.second - am) < 1e-7) {
			pa.insert(i.first);
		}
	}
	for (auto i : B) {
		if (i.second - bm > 1e-7) {
			pb.clear();
			pb.insert(i.first);
			bm = i.second;
		}
		else if (abs(i.second - bm) < 1e-7) {
			pb.insert (i.first);
		}
	}
	
	if ((pa.find(tb) != pa.end() && pb.find(ta) != pb.end())||(pa.empty()&&pb.empty()))
		cout << a << ' ' << b;
	else {
		if (b[0] == '-') {
			for (auto i:  pa) 
				cout << a << " -" << i << endl;
			for (auto i : pb) 
				cout << b << ' ' << i << endl;
		}else {
			for (auto i : pa) 
				cout << a << ' ' << i << endl;
			for (auto i : pb) 
				cout << b << " -" << i << endl;
		}
	}
	return 0;
}

vector存储照片信息

找a，b所在的照片及对存在的异性记录亲密度与寻找并记录pa，pb同步进行

#include<bits/stdc++.h>
#include<unordered_set>
using namespace std;
const int  N = 1005;
int n, m, k, sex[N];
vector<int>a[N];
void print(int v, int b){ 
	if(sex[v]) cout<< "-"; cout<<v<<" "; 
	if(sex[b]) cout << "-"; cout<<b<<endl;
}
int main(){    
	scanf("%d%d", &n, &m);    
	char t[10];    
	for (int i = 0; i < m; i++) {
		scanf("%d", &k);
		for (int j = 0; j < k; j++) {
			cin >> t;
			int y = abs(atoi(t));
			if (t[0] == '-') sex[y] = 1;
			a[i].push_back(y);
		}
	}
	char a1[10], b1[10]; 
	cin >> a1; 
	int w = abs(atoi(a1));    
	if (a1[0] == '-') sex[w] = 1;    
	cin >> b1; 
	int z = abs(atoi(b1));
	if (b1[0] == '-') sex[z] = 1; 
	double cc[1005], dd[1005];
	memset(cc, 0, sizeof(cc)); 
	memset(dd, 0, sizeof(dd)); 
	double maxa = 0, maxb = 0; 
	int fa = 0, fb = 0; 
	for (int i = 0; i < m; i++) {
		vector<int>::iterator it;  
		it = find(a[i].begin(), a[i].end(), w);   
		if (it != a[i].end()) {
			for (int j = 0; j < a[i].size(); j++) {
				int c = a[i][j];
				if (sex[w] != sex[c]) {
					cc[c] += 1.0 / (1.0 * a[i].size());    // cout<<cc[c]<<endl;                  
					maxa = max(maxa, cc[c]);
				}
			}
		}        
		it = find(a[i].begin(), a[i].end(), z);   
		if (it != a[i].end()) {
			for (int j = 0; j < a[i].size(); j++) {
				int c = a[i][j];
				if (sex[z] != sex[c]) {
					dd[c] += 1.0 / (1.0 * a[i].size());    // cout<<cc[c]<<endl;           
					maxb = max(maxb, dd[c]);
				}
			}
		}
	} if (cc[z] == maxa && dd[w] == maxb) { 
		print(w, z); 
	}
	else { 
		for (int i = 0; i < n; i++) { 
			if (maxa == cc[i]) { print(w, i); }
		}  for (int i = 0; i < n; i++) {
			if (maxb == dd[i]) { print(z, i); }
		}
	}
}

笔记：

set反向遍历：rbegin();rend()

异或：a^b

string->int转换：istringstream>>ss(str);ss>>tem;

unordered_set：只去重 不排序，速度比 set 快得多

参考：

 C++小技巧: 集合(vector, list, map, set)的反向遍历_逗神大人的博客-CSDN博客_vector反向遍历

C/C++中string和int相互转换的常用方法_AlbertS的博客-CSDN博客_c++ string转int