字节跳动笔试编程题--用户喜好

问题:

为了不断优化推荐效果，今日头条每天要存储和处理海量数据。假设有这样一种场景：我们对用户按照它们的注册时间先后来标号，

对于一类文章，每个用户都有不同的喜好值，我们会想知道某一段时间内注册的用户（标号相连的一批用户）中，有多少用户对这类文章喜好值为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

代码

#include<iostream>
#include<vector>
#include<unordered_map>
#include<algorithm>

using namespace std;

int main()
{
    int n;cin >> n;
    unordered_map<int, vector<int> > k_index;
    for (int i = 1;i <= n;i++)
    {
        int data;cin >> data;
        k_index[data].push_back(i);
    }
    //输入查询
    int q;cin >> q;
    while (q--)
    {
        int l, r, k;
        cin >> l >> r >> k;
        unordered_map<int, vector<int> >::iterator temp_it = k_index.find(k);
        if (temp_it == k_index.end())
            cout << 0 << endl;
        else
        {
            //二分查找
            auto ln = lower_bound(temp_it->second.begin(), temp_it->second.end(),l);
            auto rn = upper_bound(temp_it->second.begin(), temp_it->second.end(),r);
            cout << rn - ln << endl;
        }
    }
    return 0;
}

说明:第一次写博客是为了记录这次的题解,题解花了我一个下午的时间,

最难懂的一点是,rn - ln 是什么意思呢,这要联系前文代码的find(k),find(k)是为了找到兴趣是k的所有下标数组,然后找到max和min的下标,相减即是个数