字节跳动历届笔试题（5）

题目描述

作为一个手串艺人，有金主向你订购了一条包含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颗串珠是透明的。

思路：针对环形数组，我一般的做法是下标对长度取余，不仅代码简洁，也更容易理解，希望大家可以借鉴。对于该题，我们只需要保存每种颜色上次出现的柱子标号即可，之后判断当前位置和上一个位置的距离是否小于m，从而更新答案。

#include<stdio.h>
int n,m,c,a[10005][1005],last[55],flag[55];
int main(void){
    scanf("%d%d%d",&n,&m,&c);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i][0]);
        for(int j=1;j<=a[i][0];j++)
            scanf("%d",&a[i][j]);
    }
    int id=0,ans=0;
    for(int i=1;i<=c;i++)
        last[i]=-100000;
    while(id<2*n){
        for(int i=1;i<=a[id%n][0];i++){
            if(id-last[a[id%n][i]]<m && flag[a[id%n][i]]==0){
                flag[a[id%n][i]]=1;
                ans++;
            }
            last[a[id%n][i]]=id;
        }
        id++;
    }
    printf("%d\n",ans);
    return 0;
}