用户喜好——字节跳动2018校招大数据方向（第二批）

1.题目

[编程题]用户喜好
时间限制：C/C++ 3秒，其他语言6秒
空间限制：C/C++ 256M，其他语言512M

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

2.思路

使用hashmap，将喜好值作为key，将同一喜好值的用户组成的动态数组作为value。由于用户的标号是按照1,2,3…顺序递增的所以顺序将其存入value数组中时，value数组是从小到大排好序的，因此我们可以很方便的对其进行二分查找，通过一次二分左边界查找和一次二分右边界查找就得出了[l,r]范围内喜好度为k的用户的左右边界编号，然后右边界编号 - 左边界编号 + 1即为所求用户的个数。

3.源码

import java.util.Scanner;
import java.util.ArrayList;
import java.util.*;
 
 
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        Map<Integer,ArrayList<Integer>> map = new HashMap<>();
        for(int i = 0 ;i<n;i++){
            int key = sc.nextInt();
            if(!map.containsKey(key)){
                ArrayList<Integer> ar = new ArrayList<Integer>();
                ar.add(i+1);
                map.put(key,ar);
            }
            else{
                ArrayList arr = map.get(key);
                arr.add(i+1);
                map.put(key,arr);
            }
        }
        int q = sc.nextInt();
        for(int j=0;j<q;j++) {
            int l = sc.nextInt();
            int r = sc.nextInt();
            int k = sc.nextInt();
            if (!map.containsKey(k)) {
                System.out.println(0);
            } else {
                ArrayList<Integer> array = map.get(k);
                int left = 0;
                int right = r - 1;
                int lastleft = 0;
                int lastright = 0;
                if (array.size() < r) {
                    right = array.size() - 1;
                }
                while (left <= right) {
                    int mid = (left + right) / 2;
                    if (array.get(mid) < l) {
                        left = mid + 1;
                    } else if (array.get(mid) > r) {
                        right = mid - 1;
                    } else if (array.get(mid) <= r && array.get(mid) >= l) {
                        right = mid - 1;
                    }
 
                }
                lastleft = left;
                right = r - 1;
                if (array.size() < r) {
                    right = array.size() - 1;
                }
                while (left <= right) {
                    int mid = (left + right) / 2;
                    if (array.get(mid) < l) {
                        left = mid + 1;
                    } else if (array.get(mid) > r) {
                        right = mid - 1;
                    } else if (array.get(mid) <= r && array.get(mid) >= l) {
                        left = mid + 1;
                    }
 
                }
                lastright = right;
                int number = lastright - lastleft + 1;
                System.out.println(number);//(number);
            }
        }
    }
}

4.实际提交结果

5.经验教训

确定边界和输出时-1,+1有些混乱。