算法笔记练习 4.2 散列 问题 A: 谁是你的潜在朋友

算法笔记练习 题解合集

题目链接

题目

题目描述
“臭味相投”——这是我们描述朋友时喜欢用的词汇。两个人是朋友通常意味着他们存在着许多共同的兴趣。然而作为一个宅男，你发现自己与他人相互了解的机会 并不太多。幸运的是，你意外得到了一份北大图书馆的图书借阅记录，于是你挑灯熬夜地编程，想从中发现潜在的朋友。
首先你对借阅记录进行了一番整理，把N个读者依次编号为1,2,…,N，把M本书依次编号为1,2,…,M。同时，按照“臭味相投”的原则，和你喜欢读同一本书的人，就是你的潜在朋友。你现在的任务是从这份借阅记录中计算出每个人有几个潜在朋友。

输入
每个案例第一行两个整数N,M，2 <= N ，M<= 200。接下来有N行，第i(i = 1,2,…,N)行每一行有一个数，表示读者i-1最喜欢的图书的编号P(1<=P<=M)

输出
每个案例包括N行，每行一个数，第i行的数表示读者i有几个潜在朋友。如果i和任何人都没有共同喜欢的书，则输出“BeiJu”（即悲剧，^ ^）

样例输入
4 5
2
3
2
1

样例输出
1
BeiJu
1
BeiJu

思路

1. 数组 readers[N] 记录每个读者最喜欢的图书的编号 P，其中 readers[i] 的值代表第 i+1 个读者的 P
2. 数组 cnt_books[M+1] 统计每本书被喜欢的总数，其中 cnt_books[p] 的值代表编号为 p 的那本书被喜欢的总数
3. 读入数据的过程中更新 readers，cnt_books
4. 对每一位读者输出结果，如果 cnt_books[ readers[i] ] 的值是 1，代表这位读者喜欢的书只有他一个人喜欢，输出 BeiJu；如果值大于 1，代表有别人也喜欢这本书，输出 cnt_books[ readers[i] ] - 1

代码

#include <stdio.h>
int main(){
	int N, M, P, i;
	while (scanf("%d", &N) != EOF){
		scanf("%d", &M);
		// 记录 N 个读者最喜欢的图书的编号 P
		int readers[N];
		for (i = 0; i < N; ++i)
			readers[i] = 0; 
		// 和题目统一，cnt_books[i] 统计第 i 本书出现的次数 
		int cnt_books[M+1];
		for (i = 0; i < M+1; ++i)
			cnt_books[i] = 0;
		for (i = 0; i < N; ++i){
			scanf("%d", &P);
			readers[i] = P; 
			++cnt_books[P];
		}
		for (i = 0; i < N; ++i){
			if (cnt_books[readers[i]] == 1)
				printf("BeiJu\n");
			else
				printf("%d\n", cnt_books[readers[i]] - 1); 
		} 
	} 
	return 0;
}