欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

牛客网机试题-手串

程序员文章站 2024-03-17 14:34:46
...

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

用哈希表来存储数据,颜色的种类为key,手串的位置为value。

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(); //对串珠的限制
        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 colornum=sc.nextInt();
            for(int j=0;j<colornum;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);
    }
}