谁是你的潜在朋友

79 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第25天,点击查看活动详情

题目描述

“臭味相投”——这是我们描述朋友时喜欢用的词汇。

两个人是朋友通常意味着他们存在着许多共同的兴趣。

然而作为一个宅男,你发现自己与他人相互了解的机会并不太多。

幸运的是,你意外得到了一份北大图书馆的图书借阅记录,于是你挑灯熬夜地编程,想从中发现潜在的朋友。

首先你对借阅记录进行了一番整理,把 NN 个读者依次编号为 1,2,…,N1,2,…,N,把 MM 本书依次编号为 1,2,…,M1,2,…,M。

同时,按照“臭味相投”的原则,和你喜欢读同一本书的人,就是你的潜在朋友。

你现在的任务是从这份借阅记录中计算出每个人有几个潜在朋友。

输入格式

第一行包含两个整数 NN 和 MM。

接下来 NN 行,每行包含一个整数 PP,表示一个读者最喜欢的书籍编号。

输出格式

共 NN 行,每行输出一个整数,其中第 ii 行的整数表示第 ii 名读者的潜在朋友数量,如果没有潜在朋友则输出 BeiJu(即悲剧,^ ^)。

数据范围

2≤N,M≤2002≤N,M≤200, 1≤P≤M1≤P≤M

输入样例:

4 5
2
3
2
1

输出样例:

1
BeiJu
1
BeiJu

思路

算法1 (哈希表) 由于答案要按顺序输出,所以需要用一个数组记录每个序号的读者所阅读的书籍 遍历数组,用哈希表记录每本书籍被多少读者占用,最后根据输入顺序统计潜在朋友个数即可

注意:

  • 时间复杂度:O(n)O(n)
  • 空间复杂度:O(n)O(n)

代码

import java.util.*;
​
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(), m = sc.nextInt();
        int[] pos = new int[200];
        Map<Integer,Integer> map = new HashMap<>();
        for(int i = 0; i < n; i++){
            int a = sc.nextInt();
            pos[i] = a;
            map.put(a, map.getOrDefault(a, 0) + 1);
        }
        for(int i = 0; i < n; i++){
            if(map.get(pos[i]) > 1) System.out.println(map.get(pos[i]) - 1);
            else System.out.println("BeiJu");
        }
    }
}