手串（字节2018校招）

2.

手串

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

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 64M，其他语言128M

输入描述：

第一行输入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颗串珠是透明的。

个人思路：

统计同一种颜色出现的珠子的位置标号放入一个数组中，同时构建一个map集合，key值为颜色编号，value值为标号数组。由于他是顺时针一个一个给出的珠子，所以数组中的数字标号一定从小到大，直接遍历数组，用后一个减去前面一个和m作比较就能判断是否合理，同时这个手串是首尾相连的，还要判断最后一个和第一个的距离。

代码实现：

在数组[0]位存该数组的有效长度。

package 手串;

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

public class test {
    public static void main(String[] args) {
        int ans=0;
        Scanner scanner=new Scanner(System.in);
        int n=scanner.nextInt();
        int m=scanner.nextInt();
        int c=scanner.nextInt();
        HashMap<Integer, int[]> map=new HashMap<>();
        for(int j=1;j<=n;j++){
            int num=scanner.nextInt();
            for(int i=0;i<num;i++){
                int color=scanner.nextInt();
                if(!map.containsKey(color)){
                    int []a=new int[n+1];
                    a[0]=2;
                    a[1]=j;
                    map.put(color,a);
                }else{
                    int []arr=map.get(color);
                    arr[arr[0]++]=j;
                    map.replace(color,arr);
                }
        }
        }
        for(int i=1;i<=c;i++){
            int []arr=new int[n];
            if(map.containsKey(i)){
                arr=map.get(i);
                if(arr[0]<=2)
                    continue;
                else{
                    for(int j=1;j<arr[0]-1;j++){
                        if(arr[j+1]-arr[j]<m){
                            ans++;
                            break;
                        }
                        if(j==arr[0]-2 && arr[1]+n-arr[j+1]<m){
                            ans++;
                            break;
                        }
                    }
                }
            }
        }
        System.out.println(ans);
    }
}