用户喜好查询——2018字节前端面试编程题
为了不断优化推荐效果,今日头条每天要存储和处理海量数据。假设有这样一种场景:我们对用户按照它们的注册时间先后来标号,对于一类文章,每个用户都有不同的喜好值,我们会想知道某一段时间内注册的用户(标号相连的一批用户)中,有多少用户对这类文章喜好值为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
这道题主要问题是使用顺序扫描用户喜爱度数组的时候会花费比较长的时间
所以使用hashmap将该喜好值为k和对应的用户id的list进行存储
let SearchGroup=[ ];
let usernumber=readline(),
likearray=readline().split(' '), //用户对应喜好,转化为数组
SearchN=readline(); //第三行输入,查询组数
for(let i=0;i<SearchN;i++){
SearchGroup[i]=readline().split(' ');
}
//样例添加完后生成arr=[,[0],[1],[2,3],,[4]]
//顺序遍历likearray数组,对每个item项进行数组创建和数组下标的添加
let arr=[ ];
likearray.forEach((item,index)=>{
if(arr[item]==undefined){ //找不到
arr[item]=[ ]; //新建数组
}
arr[item].push(index); //arr是处理之后的数组
});
for(let j=0;j<SearchN;j++){
let start=SearchGroup[j][0]-1,
end=SearchGroup[j][1]-1,
value=SearchGroup[j][2],
result=0; //一堆初始化
if(arr[value]==undefined){ //数组value值为空
console.log(0);
}else{
arr[value].forEach(e=>{
if(e>=start&&e<=end){
result++;
}
})
print(result);
}
}
//注意forEach的使用
/*
js中=>的作用
(x) => x + 6
function(x){
return x + 6; }
*/
整理自牛客网
链接:https://www.nowcoder.com/questionTerminal/e9d0a72fb08044efb8263946f405ec4b