图算法：推荐潜在好友 投机取巧法 201805

问题转载自：Lambda OJ 201805

问题

社交网络可以用一个图来描述。假定有N个使用者，每个使用者可以用一个节点表示，如果使用者m和使用者n已经是线上朋友，表示两个节点已经连接，即图中包含连接m和n的边。在这个练习中，我们将为使用者推荐可能的朋友。

给定表示一个社交网络的图，包含N个节点和M条边。如果节点m和n不直接相连，但与K个以上（包括K个）的相同节点连接，我们可以认为他们可能认识，请给出指定用户的潜在好友（暂时还未直接连接的节点）。对于给定用户，请将其所有潜在好友按照他们共同好友数量由多到少排列输出，对于共同好友数量相同的情况，请按照序号从小到大排列输出。

输入格式

输入的第一行是三个整数N，K，U。其中N表示社交网络中用户的个数，即网络节点的个数。K表示共同好友数不少于K的两个用户才互相算作潜在好友。U表示需要输出标号为U的用户的潜在好友，对应矩阵行/列标号为U的节点，U从0开始计。
输入的第二行到第N+1行是用户之间的图的邻接矩阵：0代表不是好友，1代表是好友。

输出格式

请在一行中，输出用户U所有可能认识的人（共同好友数量不少于K并且目前还不是好友的人）的编号（即对应的矩阵的行列序号，从0开始计数），序号之间用空格隔开。按照共同好友数量由多到少排列输出，对于共同好友数量相同的潜在朋友，请按照序号从小到大排列输出。

输入样例

6 1 3
0 1 0 1 0 1 
1 0 0 0 0 1 
0 0 0 0 0 1 
1 0 0 0 0 1 
0 0 0 0 0 1 
1 1 1 1 1 0

输出样例

1 2 4

提示

图的邻接矩阵对角线元素没有意义，且为对称阵。
实测数据量大概3000，排序不能使用qsort库函数。

题解

技巧在于：不需要把所有人的关系都存下来，将空间复杂度降到O(n)。
为了看第u个人的好友，只需每读入一行（表示某个人的好友关系），看是否与他是好友，如果是，就把这一行对应的人全部好友关系记下来，依次加入一个额外的一维数组中。最后这个额外数组每项就存下了第u人与其他人的共同好友数目总和。
当然，为了排除已经是好友的人，这些项应该在发现关系时置为一个足够小的数，比如-n；这样保证最后加到一起是负的值。本人同理。
最后只需对这个数组排序、依次输出直到共同好友数不足要求为止。

原创解法：

#include <stdio.h>
#define MAX 5000
struct human{
	int id;
	int frnd;
};
/为了写这个Qsort还是很费心的
void swap(struct human s[],int i,int j)
{
	struct human temp;
	temp=s[i];
	s[i]=s[j];
	s[j]=temp;
}
int shouldswap(struct human *a,struct human *b){
	if(a->frnd<b->frnd)return 1;
	else if(a->frnd>b->frnd)return -1;
	//优先按照共同好友数目排序，再按照每个人的序号排序
	else return a->id-b->id;
}
void QSort(struct human s[MAX],int low,int high)
{
	int i;
	int last;       
	if(low<high)    
	{
		last=low;   
		for(i=low+1;i<=high;i++)
		{
			if(shouldswap(s+i,s+low)<0)
				swap(s,++last,i);
		}
		swap(s,last,low);
		QSort(s,low,last-1);
		QSort(s,last+1,high);
	}
}
/*废掉的comp
int comp(const void*a,const void*b){
	if(((struct human*)a)->frnd<((struct human*)b)->frnd)return 1;
	else if(((struct human*)a)->frnd>((struct human*)b)->frnd)return -1;
	else return ((struct human*)a)->id-((struct human*)b)->id;
}
*///

int main(){
	int n,m_min,find_id;
	int v[MAX];
	struct human hu[MAX];
	scanf("%d%d%d",&n,&m_min,&find_id);
	for(int i=0;i<n;i++){
		hu[i].id=i;
		hu[i].frnd=0;
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			scanf("%d",v+j);
		}
		if(v[find_id]){
			hu[i].frnd=-n;

			for(int k=0;k<i;k++){
				hu[k].frnd+=v[k];
			}
			for(int k=i+1;k<n;k++){
				hu[k].frnd+=v[k];
			}
		}
	}
	hu[find_id].frnd=-n;
	QSort(hu,0,n-1);
	n=0;
	while(hu[n].frnd>=m_min){
		printf("%d ",hu[n].id);
		n++;
	}
}