今日头条2018校招后端方向(用户喜好)c++

题目描述

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

题解:

一开始用的vector型数组,结果k值太大,直接段错误。

正解:利用c++ STL库中 multimap,multimap提供了可以一种可以有重复键值的STL map类型。其插入方式和map相似。不会用的先自行百度吧。 

这里需要解释下边代码中:ret=equal_range(k):函数的返回值是一个pair,分别存放相同键k的迭代器区间。

插入以用户喜好值为key,以用户标号为value。剩下的就是遍历相同键k的区间。

#include<bits/stdc++.h>
using namespace std;
#define mk make_pair
typedef long long ll;
multimap <int,int> mp;
int n;
int a,q;
int l,k,r,ans;
int main(void)
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a);
        mp.insert(mk(a,i));
    }
    scanf("%d",&q);
    for(int i=0;i<q;i++)
    {
        scanf("%d%d%d",&l,&r,&k);
        auto ret=mp.equal_range(k);
        ans=0;
        for(auto it=ret.first;it!=ret.second;it++)
        {
            if(it->second>=l&&it->second<=r)
                ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Yang_1998/article/details/82252327