标签搜索

目 录CONTENT

文章目录

字节--用户喜好.md

小小城
2021-08-22 / 0 评论 / 0 点赞 / 5 阅读 / 2,134 字 / 正在检测是否收录...
温馨提示:
本文最后更新于 2022-05-02,若内容或图片失效,请留言反馈。部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

字节--用户喜好

@[toc]

一、题目描述

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

二、分析

哎,字节爸爸的题永远不要想的太简单,题中已经说到---海量数据,暴力求解只能超时,所以需要另辟捷径

  •  这道题我们可以把用户的喜好值和用户的id封装在一起,Node
typedef struct __tagNode
{
    int val;
    int index;
    __tagNode(int x, int y)
    {
        val = x;
        index = y;
    }
}Node;
  •  Node中一个是喜好值val,一个是index当前索引
  •  输入数组设置为:vector<node>vec,然后按照val对vec进行排序,这样就会把同一个val的相邻。
class less_to_greater
{
    public:
    bool operator()(Node &x, Node &y)
    {
        return x.val < y.val;
    }
};
  •  然后用一个map记录每个不同的val的初始索引当查询一个val的时候,只需要根据map中val的初始索引,在vec中向后查找,统计index在给定的left值和right范围内的个数即可,这么做为了避免每次查询一个val,就会遍历很多数

三、代码

#include<iostream>
#include<vector>
#include<unordered_map>
#include<algorithm>
  
using namespace std;
//封装Node
typedef struct __tagNode
{
    int val;//用户喜好值
    int index;//用户下标
    __tagNode(int x, int y)
    {
        val = x;
        index = y;
    }
}Node;

//自定义比较方式:以用户的喜好值排序
class less_to_greater
{
    public:
    bool operator()(Node &x, Node &y)
    {
        return x.val < y.val;
    }
};

int main()
{
    int n;
    cin >> n;
    vector<Node>vec(n, Node(0, 0));
    //构造数组
    for(int i = 0; i < n; ++i)
    {
        cin >> vec[i].val;
        vec[i].index = i;
    }
    
    // 排序
    sort(vec.begin(), vec.end(), less_to_greater());
    
    // 记录每个喜好对应索引的位置
    //如果存在多个相同的喜好值,以第一个为准
    unordered_map<int, int>ma;
    for(int i = 0; i < n; ++i)
    {
        if(ma.count(vec[i].val) < 1)
            ma[vec[i].val] = i;
    }
    
    // 开始查询
    int q;
    cin >> q;
    while(q)
    {
        int l, r, k;
        cin >> l >> r >> k;
        //排除不合法情况和没有k喜好值的情况
        if(l > r || ma.count(k) < 1)
        {
            cout << 0 << endl;
            --q;
            continue;
        }

		//--的原因是房间id是从1开始的
        --l;
        --r;

		//获取喜好值为k的索引位置
        int i = ma[k];
        //保存结果
        int res = 0;
        //因为可能不只i号用户的喜好值为k,上面保存的时候只以第一个为准
        //如果i位置在l和r之间更新结果
        while(i < n && vec[i].val == k)
        {
            if(vec[i].index >= l && vec[i].index <= r)
                ++res;
            ++i;
        }
        cout << res << endl;
        --q;
    }
    return 0;
}
0

评论区