【算法】字节跳动2018校招Android方向（第二批）笔试题：用户喜好

题目
牛客题目链接

为了不断优化推荐效果，今日头条每天要存储和处理海量数据。假设有这样一种场景：我们对用户按照它们的注册时间先后来标号，对于一类文章，每个用户都有不同的喜好值，我们会想知道某一段时间内注册的用户（标号相连的一批用户）中，有多少用户对这类文章喜好值为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

思路：

• 第一想法是暴力一个一个遍历，但是牛客判定超时只过了50%；
• 后来在网上找到方法有人使用Map，我就借鉴过来了，意思就是把 让喜好值作为Map的key ，创建一个ArrayList集合存下喜好值为key的人的位置，然后让这个 ArrayList作为Map的value 。

以下是代码

import java.util.*;
import java.lang.*;

/**
 * @author LZY
 */
public class Main{
    public static void main(String[] args){
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();
        Map<Integer,List<Integer>> map=new HashMap();
        List<Integer> like=new ArrayList<>();
        for(int i = 0;i < n; i++){
            like.add(sc.nextInt());
        }
         for(int i = 0;i < n; i++){
             //如果map中不存在该like为键的键值对，则插进去一个新的键值对
            if(!map.containsKey(like.get(i))){
                List<Integer> list =new ArrayList<>();
                //存储喜好值的位置（喜好值为like.get(i)的人的位置）
                list.add(i+1);
                //把对应的喜好值和是该喜好值的人分别存key里和list集合中
                map.put(like.get(i),list);
            }else{
                map.get(like.get(i)).add(i+1);
            }
        }
        int q=sc.nextInt();
        for(int i = 0;i < q; i++){
            int l=sc.nextInt();
            int r=sc.nextInt();
            int key=sc.nextInt();
            int  answer=0;
               //回车
            sc.nextLine();
             List<Integer> res=map.get(key);
            if(res!=null){
                 for(int result:res){
                     if(result>=l&&result<=r){
                          answer++;
                     }
                }
            }
        //输出答案
            System.out.println(answer);
        }
    }
}