用户喜好牛客网- 今日头条

题目

[编程题] 用户喜好

时间限制：3秒

空间限制：262144K

为了不断优化推荐效果，今日头条每天要存储和处理海量数据。假设有这样一种场景：我们对用户按照它们的注册时间先后来标号，对于一类文章，每个用户都有不同的喜好值，我们会想知道某一段时间内注册的用户（标号相连的一批用户）中，有多少用户对这类文章喜好值为k。因为一些特殊的原因，不会出现一个查询的用户区间完全覆盖另一个查询的用户区间(不存在L1<=L2<=R2<=R1)。

输入描述:

输入：第1行为n代表用户的个数第2行为n个整数，第i个代表用户标号为i的用户对某类文章的喜好度第3行为一个正整数q代表查询的组数第4行到第（3+q）行，每行包含3个整数l,r,k代表一组查询，即标号为l<=i<=r的用户中对这类文章喜好值为k的用户的个数。数据范围n <= 300000,q<=300000 k是整型

输出描述:

输出：一共q行，每行一个整数代表喜好值为k的用户的个数

输入例子1:

5
1 2 3 3 5
3
1 2 1
2 4 5
3 5 3

输出例子1:

1
0
2

例子说明1:

样例解释:
有5个用户，喜好值为分别为1、2、3、3、5，
第一组询问对于标号[1,2]的用户喜好值为1的用户的个数是1
第二组询问对于标号[2,4]的用户喜好值为5的用户的个数是0
第三组询问对于标号[3,5]的用户喜好值为3的用户的个数是2
！

思路：

在给出的一个数组中，在l和r之间，有多少个k。如何直接遍历查找的话会超时。其实我们可以把用户
喜好即K值作为一个标记，然后里面存储数组的index，接着判断index是否在l和r之间就行了。

使用链表数组来做的，k值作为数组下标，但是后来发现K值很大，会出现数组越界
题目没有给出K值的限制，如果直接使用数组下标来作为k值的标记，那么会出现数组越界错误。所以，我们使用
HashMap，用户爱好作为key，value为相同爱好的数组的index，我们使用一个单链表来存储value。

import java.util.*;
public class Main{
    /**
     * 创建一个HashMap的对象
     * 声明一个类型为 HashMap的引用变量hashMap,注意键值对为<Integer,Node>
     *  把新的HashMap的对象赋值给hashMap这个引用变量（即引用变量指向这个对象）
     */
    public static HashMap<Integer,Node> hashMap=new HashMap<>();

    /**
     * 创建一个单链表
     */
    static class Node{
        private int position;
        private Node Next;
        public  Node(int position )   {
            this.position= position;
        }

        /**
         * 该链表中增加node节点的方法
         * 如果当前节点的Next为null，则把新的节点newNode赋值给当前节点
         * 否则，在当前节点的Next指向下一个节点newNode
         * @param newNode
         */
        public void addNode(Node newNode){
            if(this.Next==null){
                this.Next= newNode;
            }else{
                this.Next.addNode(newNode);
            }
        }
    }

    public static void main(String[] args){
        /**
         * 通过new Scanner(System.in)创建一个Scanner对象，控制台会一直等待输入，直到敲回车键结束，把所输入的内容传给Scanner，作为扫描对象。
         * 把扫描对象赋值给in这个引用变量
         * 注意 nextInt()与nextLine()区别
         */
        Scanner in = new Scanner(System.in);
        int n,q,l,r,k,num;
        n= in.nextInt();
        for(int i=1;i<=n;i++){
            num= in.nextInt();
            Node node =new Node(i);//创建不同的Node对象给,并赋值给引用变量node

            /**
             * hashMap.get(num)
             * num相当于hashMap<key,value>中的key
             * hashMap.get(num)  获取对应的value值
             */
            if(hashMap.get(num)!=null){//判断hashmap是否存在该用户喜好，存在就添加到链尾
                hashMap.get(num).addNode(node);
            }else if(hashMap.get(num)==null){
                hashMap.put(num,node);
            }
        }
        q=in.nextInt();
        for(int i=0;i<q;i++){
            l= in.nextInt();
            r= in.nextInt();
            k= in.nextInt();
            int count = 0;
            Node root= hashMap.get(k);
            while(root!= null&&root.position<=r){
                if(root.position>=l && root.position<=r){
                    count++;
                }
                root=root.Next;//指向下一个节点
            }
            System.out.println(count);
        }
    }
}

代码已经分析的很清楚了，说明的是，执行完语句后

 Scanner in = new Scanner(System.in);
        int n,q,l,r,k,num;
        n= in.nextInt();
        for(int i=1;i<=n;i++){
            num= in.nextInt();
            Node node =new Node(i);//创建不同的Node对象给,并赋值给引用变量node

            /**
             * hashMap.get(num)
             * num相当于hashMap<key,value>中的key
             * hashMap.get(num)  获取对应的value值
             */
            if(hashMap.get(num)!=null){//判断hashmap是否存在该用户喜好，存在就添加到链尾
                hashMap.get(num).addNode(node);
            }else if(hashMap.get(num)==null){
                hashMap.put(num,node);
            }
        }

哈希表及链表情况如图所示：

代码参考自：https://blog.csdn.net/callmeMrLu/article/details/81917158

用户喜好 牛客网- 今日头条

猜你喜欢

用户喜好牛客网- 今日头条