Zoom 8.10 笔试第二题股票推荐 当时没做出来

import java.util.*;

public class Main{
    static Map<String, Set<String>> map = new HashMap<>();//股民姓名为Key,Set集合为订阅股票 Value
    static Set<String> resultSet = new LinkedHashSet<>();//推荐的股票集合
    static List<String> nameList = new LinkedList<>();//股民姓名链表 方便遍历

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int count = Integer.parseInt(scanner.nextLine());//输入操作次数

        while(count>0) {
            count--;
            String stepStr = scanner.nextLine();
            String[] steps = stepStr.split(" ");//每一步操作的 steps 例如 1 Alice 2  注意此处不包含插入的订阅股票neirong内容
            if(steps[0].equals("1")|| steps[0].equals("2")){
                //插入或者查询操作
                if(steps[0].equals("1")){

                    //插入
                    Set<String> set = new LinkedHashSet<>();
                    //读取插入
                    String insertStr = scanner.nextLine();
                    String[] insertContent = insertStr.split(" ");                     
                    for (String content : insertContent) {//加入集合
                        set.add(content);
                    }
                    //加数据 加名字
                    map.put(steps[1],set);
                    nameList.add(steps[1]);
                } else if(steps[0].equals("2")){
                    //查找
                    if(!map.containsKey(steps[1])){//不存在输出error
                        System.out.println("error");
                        continue;
                    }else {
                        //返回个数
                        int searchNum = search(steps[1]);
                        System.out.println(searchNum);
                    }
                }
            }

        }


    }
    public static int search(String name){

        Set<String> curSet = map.get(name);

        for (String otherName : nameList) {
            if(!name.equals(otherName)){//遍历查询与当前的用户 订阅有交集的股民
                Set<String> otherSet = new LinkedHashSet<>(map.get(otherName));//切记new
                Set<String> otherNoDelSet = new LinkedHashSet<>(map.get(otherName));//切记new

                otherSet.retainAll(curSet);
                if(otherSet.size()==0){//0说明无交集 无法推荐
                    continue;
                }else {
                    //删除交集加入到总结果集合
                    otherNoDelSet.removeAll(otherSet);
                    resultSet.addAll(otherNoDelSet);
                }

            }

        }
        int nums = resultSet.size();
        resultSet.clear();
        return nums;


    }    

#zoom校招##做完zoom2023秋招笔试,人麻了#
全部评论
可以看看你的第一题么,看他们C++好短
点赞 回复 分享
发布于 2022-08-10 21:24
你这个每次都要遍历一下,如果股票总数量级为M,人的数量级为N的话,每次去和其他人取交集,复杂度已经是O(M*N)了,每个人都这样做一下,复杂度为O(M*N^2)了,你这时间能过得去么
点赞 回复 分享
发布于 2022-08-10 22:47

相关推荐

牛舌:如果我不想去,不管对方给了多少,我一般都会说你们给得太低了。这样他们就会给下一个offer的人更高的薪资了。
点赞 评论 收藏
分享
评论
2
4
分享
牛客网
牛客企业服务