区间查找

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

import java.util.*;
//Definition for an interval.
class Interval {
      int index;
      int val;
      Interval() { index = 0; val = 0; }
      Interval(int s, int e) { index = s; val = e; }
  }

public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        //int[] val = new int[n];
        List<Interval> inter = new ArrayList<Interval>();
        for(int i = 1; i<=n; i++){
             //val[i] = sc.nextInt();
            int tmp = sc.nextInt();
            inter.add(new Interval(i, tmp));
        }
        Collections.sort(inter, new Comparator<Interval>(){
            @Override
            public int compare(Interval s, Interval e){
                return s.val-e.val;
            }
        });
        int q = sc.nextInt();
        int[][] wait = new int[q][3];
        for(int j = 0; j < q; j++){
             wait[j][0] = sc.nextInt();
             wait[j][1] = sc.nextInt();
             wait[j][2] = sc.nextInt();
        }
        for(int k = 0; k < q; k++){
            System.out.println(find(inter, wait[k][0], wait[k][1], wait[k][2]));
        }
    }
    public static int find(List<Interval> inter, int s, int e, int tar){
        int res = 0;
        int down = erfen(inter, 0, inter.size(), tar);
        if(down == -1)
            return res;
        int up = down+1;
        while(down>=0 && inter.get(down).val == tar){
            Interval tmp = inter.get(down);
            if(tmp.index >= s && tmp.index <= e )
                res++;
            down--;
        }
         while(up < inter.size() && inter.get(up).val == tar){
            Interval tmp = inter.get(up);
            if(tmp.index >= s &&tmp.index <= e )
                res++;
            up++;
        }
        return res;
    }
    public static int erfen(List<Interval> inter, int lo, int hi, int tar){
       if(lo>=hi)
           return lo;
       int mid = (lo + hi)/2;
       if(inter.get(mid).val > tar){
           return erfen(inter,lo, mid, tar);
        }
        if(inter.get(mid).val < tar){
            return erfen(inter, mid+1, hi, tar);
        }
        else
            return mid;
}
}