首页 > 试题广场 >

手串

[编程题]手串
  • 热度指数:798 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 64M,其他语言128M
  • 算法知识视频讲解

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

public class Main{
    public static void main(String[] args){
        Scanner scanner=new Scanner(System.in);
        int n=scanner.nextInt();
        int m=scanner.nextInt();
        int c=scanner.nextInt();
        HashMap<Integer, HashSet<Integer>> map=new HashMap<>();
        for(int i=1; i<n+1; i++){
            int colors=scanner.nextInt();
             for(int j=0; j<colors; j++){
                 int cid=scanner.nextInt();
                if(!map.containsKey(cid))
                    map.put(cid, new HashSet<Integer>());
                map.get(cid).add(i);                 
             }
        }
        int ans=0;
        for(int i=0; i<c+1; i++){
            if(count(n, m, map.getOrDefault(i, new HashSet<Integer>())))
                ans++;
        }
        System.out.println(ans);
    }   
    
    private static boolean count(int n, int m, HashSet<Integer> set){
        for(int i:set){
            for(int j:set){
                if(i==j) continue;
                int min=i, max=j;
                if(i>j){
                    int t=max; max=min; min=t;
                }
                if(max-min<=m-1||n-max+min<=m-1)
                    return true;
            }
        }
        return false;
    }        
    
}


以上为Java代码,
易错点,count子函数中,当i>j的时候不能直接交换i和j的值,因为这样会修改set中的内容。
另外,HashMap中的set其实也可以用list代替,用list就可以指定位置了,循环的时候还可以指定i<j之类的,避免同样的颜色对检查两遍。
发表于 2020-04-30 15:53:07 回复(0)