字节跳动笔试题总结

题目1

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

时间限制：3秒 空间限制：262144K

解法：
本题如果用两重循环去查找答案会很简单，但是通过率只有50%，所以O(n ^ 2)的方法是行不通的。这里有个结论就是当时间限制为1s时，数据量大概是10 ^ 8 ~10 ^ 9，这里时间限制为3s，而n和q最大可以为300000，所以 O(n^ 2)会超出时间限制的。
这里面的解法就是用一个HashMap，key保存每个数字，value是一个链表，保存的是每个数字对应的下标，这样我们每次查数据只需要从key对应的链表中去查询出现的次数，大大加快了查询速度。
Code如下：

import java.util.*;

public class Main {
    public static int process(int l, int r, int k, HashMap<Integer, List<Integer>> map) {
        int res = 0;
        List<Integer> list = map.get(k);
        if (list != null) {
            for (Integer cur : list) {
                if (cur >= l && cur <= r) {
                    res++;
                }
            }
        }
        return res;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }

        HashMap<Integer, List<Integer>> map = new HashMap<>();

        for (int i = 0; i < n; i++) {
            if (!map.containsKey(arr[i])) {
                List<Integer> list = new ArrayList<>();
                list.add(i + 1);
                map.put(arr[i], list);
            } else {
                List<Integer> list = map.get(arr[i]);
                list.add(i + 1);
            }
        }

        int q = sc.nextInt();
        for (int i = 0; i < q; i++) {
            int l = sc.nextInt();
            int r = sc.nextInt();
            int k = sc.nextInt();
            System.out.println(process(l, r, k, map));
        }
    }
}

题目2

链接：https://www.nowcoder.com/questionTerminal/429c2c5a984540d5ab7b6fa6f0aaa8b5
来源：牛客网
题目描述：
作为一个手串艺人，有金主向你订购了一条包含n个杂色串珠的手串——每个串珠要么无色，要么涂了若干种颜色。为了使手串的色彩看起来不那么单调，金主要求，手串上的任意一种颜色（不包含无色），在任意连续的m个串珠里至多出现一次（注意这里手串是一个环形）。手串上的颜色一共有c种。现在按顺时针序告诉你n个串珠的手串上，每个串珠用所包含的颜色分别有哪些。请你判断该手串上有多少种颜色不符合要求。即询问有多少种颜色在任意连续m个串珠中出现了至少两次。

输入描述:
第一行输入n，m，c三个数，用空格隔开。(1 <= n <= 10000, 1 <= m <= 1000, 1 <= c <= 50) 接下来n行每行的第一个数num_i(0 <= num_i <= c)表示第i颗珠子有多少种颜色。接下来依次读入num_i个数字，每个数字x表示第i颗柱子上包含第x种颜色(1 <= x <= c)

输出描述:
一个非负整数，表示该手链上有多少种颜色不符需求。
示例1
输入
5 2 3
3 1 2 3
0
2 2 3
1 2
1 3
输出
2
说明
第一种颜色出现在第1颗串珠，与规则无冲突。
第二种颜色分别出现在第 1，3，4颗串珠，第3颗与第4颗串珠相邻，所以不合要求。
第三种颜色分别出现在第1，3，5颗串珠，第5颗串珠的下一个是第1颗，所以不合要求。
总计有2种颜色的分布是有问题的。
这里第2颗串珠是透明的。

解法：
该题初次看上去较为复杂，分析之后主要流程就是构建一个HashMap，key存颜色对应的数字，value存颜色分布在串珠的位置。以题目的测试用例为例子，
5 2 3
3 1 2 3
0
2 2 3
1 2
1 3
构建出来的哈希表就是
1 -> 0 3 4
2 -> 1 3 4
3 -> 1 3 5
如以此来，再去遍历这个哈希表的每一个队列，如果发现不符合金主爸爸的要求，就error++，有多少个队列不符合要求，error便是多少。
这里有一个坑，就是遍历每个队列的时候，只要发现了不符合之后就要break。
具体Code如下：

import java.util.List;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt(); //手串数
        int m = sc.nextInt(); //任何连续的m个串珠至多出现一次
        int c = sc.nextInt(); //颜色总数

        //每种颜色对应的位置
        HashMap<Integer, ArrayList<Integer>> map = new HashMap<>();

        //先加入空的队列
        for (int i = 1; i <= c; i++) {
            map.put(i, new ArrayList<Integer>());
        }

        //输入颜色时，以颜色为key，手串的位置为value填充
        for (int i = 1; i <= n; i++) {
            int num_i = sc.nextInt();
            for (int j = 0; j < num_i; j++) {
                int num = sc.nextInt();
                ArrayList<Integer> list = map.get(num);
                list.add(i);
                map.put(num, list);
            }
        }

        //错误的颜色数
        int error = 0;

        //遍历每个颜色对应的队列
        for (int i = 1; i <= c; i++) {
            List<Integer> list = map.get(i); //获取队列
            for (int j = 0; j < list.size(); j++) {
                if (j < list.size() - 1 && list.get(j + 1) - list.get(j) < m) {
                    error++;
                    break; //因为只需要找出错误的颜色种类，所以一旦找到就要跳出循环
                } else if (j == list.size() - 1 && list.get(0) + n - list.get(j) < m) {
                    error++;
                    break;
                }
            }
        }
        System.out.println(error);
    }
}