L2-028 秀恩爱分得快 (25分)

L2-028 秀恩爱分得快 (25分)
古人云：秀恩爱，分得快。

互联网上每天都有大量人发布大量照片，我们通过分析这些照片，可以分析人与人之间的亲密度。如果一张照片上出现了 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

#include<stdio.h>
#include<stdlib.h>
bool isMan[1001] = { false };   //记录性别，由于-0的存在不能通过正负来判断性别
double relation[2][1001] = { 0.0 }; //[0][i]代表a与i的亲密度 [1][i]代表b与i的亲密度

int main()
{
	int N, M, K, a, b, s;
	bool man;
	char ch;
	int p[500];
	scanf("%d %d", &N, &M);
	
	int index[M];       //记录每一张图片中人的总数
	int* (photo[M]);    //记录每一张图片
	
	for (int i = 0; i < M; i++)
	{
		scanf("%d", &K);
		photo[i] = (int*)malloc(sizeof(int) * K);   //为每张图片分配空间
		index[i] = K;
		getchar();
		for (int j = 0; j < K; j++)
		{
			man = true;
			s = 0;
			while ((ch = getchar()) != ' ' && ch != '\n')
			{
				if (ch == '-')man = false;
				else s = s * 10 + ch - 48;
			}
			isMan[s] = man;
			photo[i][j] = s;
		}
	}
	
	int ans1[N];
	int ans2[N];
	int ai = 0, bi = 0;
	double ah = 0, bh = 0;
    if(ch==' ')getchar();
    for(int i=0;i<2;i++){
	    s = 0;
	    man = true;
    	while ((ch = getchar()) != ' ' && ch != '\n')
	    {
		    if (ch == '-')man = false;
		    else s = s * 10 + ch - 48;
	    }
	    isMan[s] = man;
	    if(i==0)a = s;else b=s;
    }
	
    
	for (int i = 0; i < M; i++)     //在所有图片中找包含a或b的图片，然后计算他们与其他人的亲密度
	{
		for (int j = 0; j < index[i]; j++)
			if (photo[i][j] == a || photo[i][j] == b)
			{
				s = photo[i][j];
				for (int k = 0; k < index[i]; k++)
					if (isMan[photo[i][k]] != isMan[s]&&photo[i][k]!=s)
					{
						relation[s == a ? 0 : 1][photo[i][k]] += 1.0 / index[i];
					}
			}
	}
	
	for (int i = 0; i < N; i++)
	{
		if (relation[0][i] > ah) { ah = relation[0][i]; ai = 0; ans1[ai++] = i; }
		else if (relation[0][i] == ah) { ans1[ai++] = i; }

		if (relation[1][i] > bh) { bh = relation[1][i]; bi = 0; ans2[bi++] = i; }
		else if (relation[1][i] == bh) { ans2[bi++] = i; }
	}

	if (relation[0][b] == ah && relation[1][a] == bh)printf("%s%d %s%d", isMan[a]?"":"-",a, isMan[b]?"":"-",b);
	else
	{
		for (int i = 0; i < ai; i++)printf(i == 0 ? "%s%d %s%d" : "\n%s%d %s%d", isMan[a]?"":"-",a, isMan[ans1[i]]?"":"-",ans1[i]);
		for (int i = 0; i < bi; i++)printf("\n%s%d %s%d", isMan[b]?"":"-",b, isMan[ans2[i]]?"":"-",ans2[i]);
	}

}