用户喜好

（java实现）

---

题目描述：

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

说明

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

---

问题分析：

思路一：直接暴力法，容易超时。故需要考虑其他算法。
但是有个问题，如果把查询那一部分单独写成函数（如思路一实现2.0）却可以通过，不写成函数（1.0），反而过不了？目前还没搞清楚是为什么。
比较图片中的第一、第二行：从运行结果可以看出，第一行（2.0）运行状态是“答案正确”，第二行（1.0）却运行超时。从占用内存方面来看，第一行（2.0）占用“157068K”，相比于第二行（1.0）的“31340K”来说，增加将近5倍的内存（也可能是函数的调用、查找，使用到了栈，故消耗的内存比较大）。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-mZbDuxbC-1618673570648)(https://uploadfiles.nowcoder.com/images/20210323/4479402_1616511673279/273AD2E52B166F6D289FA918D4F54F14 “图片标题”)]

思路二：符号表应用，把喜好值作为key，对应该喜好值的人的集合作为value
比如查询5的时候，直接找到所有喜好值为5的人的集合，遍历集合中哪些人在范围内。
使用：Map<Integer,List>

相关知识：

Map<Integer,List> userLike = new HashMap<>();

List<Integer> list = userLike.get(key);
if (list!=null)
{
	for (Integer val : list)
	{
		if (val>=left && val<=right)
			count++;
	}
}

---

参考代码：

思路一实现（1.0）：

import java.util.*;
public class Main {
        
    public static void main(String[] args)
    {
        Scanner input = new Scanner(System.in);
        while (input.hasNext())
        {
            int n = input.nextInt();
            int[] userLike = new int[n+1];
            for (int i=1; i<=n; i++)
            {
                userLike[i] = input.nextInt();
            }
            int m = input.nextInt();
            for (int i=0; i<m; i++)
            {
                int left = input.nextInt();
                int right = input.nextInt();
                int key = input.nextInt();
                int count = 0;
                for (int j=left; j<=right; j++)
                {
                    if (key == userLike[j])
                        count++;
                }
                System.out.println(count);
            }
        }
    }
}

思路一实现（2.0）：

import java.util.*;
public class Main {
        
    public static void main(String[] args)
    {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int[] userLike = new int[n+1];
        for (int i=1; i<=n; i++)
        {
            userLike[i] = input.nextInt();
        }
        int m = input.nextInt();
        int left, right, key;
        int[] res = new int[m];
        for (int i=0; i<m; i++)
        {
            left = input.nextInt();
            right = input.nextInt();
            key = input.nextInt();
            res[i] = check(left,right,key,userLike);
        }
        for (int i=0; i<m; i++)
        {
            System.out.println(res[i]);
        }
    }
    
    public static int check(int left, int right, int key, int[] userLike)
    {
        int count = 0;
        for (int j=left; j<=right; j++)
        {
            if (key == userLike[j])
                count++;
        }
        return count;
    }
}

思路二实现：

import java.util.*;
import java.util.List;
public class Main {
        
    public static void main(String[] args)
    {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        Map<Integer,List<Integer>> userLike = new HashMap<>();
        int key = 0;
        for (int i=1; i<=n; i++)
        {
            key = input.nextInt();
            if (!userLike.containsKey(key))
            {
                List<Integer> list = new LinkedList<>();
                list.add(i);
                userLike.put(key, list);
            }
            else
            {
                List<Integer> list = userLike.get(key);
                list.add(i);
            }
        }
        int m = input.nextInt();
        int left, right, count;
        for (int i=0; i<m; i++)
        {
            left = input.nextInt();
            right = input.nextInt();
            key = input.nextInt();
            count = 0;
            List<Integer> list = userLike.get(key);
            if (list!=null)
            {
                for (Integer val : list)
                {
                    if (val>=left && val<=right)
                        count++;
                }
            }
            System.out.println(count);
        }
    }
}