字节跳动笔试题之手串

作为一个手串艺人,有金主向你订购了一条包含n个杂色串珠的手串——每个串珠要么无色,要么涂了若干种颜色。为了使手串的色彩看起来不那么单调,金主要求,手串上的任意一种颜色(不包含无色),在任意连续的m个串珠里至多出现一次(注意这里手串是一个环形)。手串上的颜色一共有c种。现在按顺时针序告诉你n个串珠的手串上,每个串珠用所包含的颜色分别有哪些。请你判断该手串上有多少种颜色不符合要求。即询问有多少种颜色在任意连续m个串珠中出现了至少两次。
字节跳动笔试题之手串_第1张图片字节跳动笔试题之手串_第2张图片
AC代码:

import java.util.*;
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();
        Map> map = new HashMap<>();
        for(int i = 0;i < c;i++) {
        	map.put(i + 1, new ArrayList());
        }
        for(int i = 0; i < n;i++) {
        	int length = scanner.nextInt();
        	if(length != 0) {
        		for(int j = 0;j < length;j++) {
            		List list = map.get(scanner.nextInt());
            		list.add(i);
            	}
        	}
        }
        int count = 0;
        for(List list : map.values()) {
        	if(list.size() > 1) {
        		boolean flag = false;
            	int size = list.size();
            	for(int i = 0;i < size - 1;i++) {
            		if(list.get(i + 1) - list.get(i) < m) {
            			count++;
            			flag = true;
            			break;
            		}
            	}
            	if(!flag) {
            		if(list.get(0) + n - list.get(size - 1) < m) {
            			count++;
            		}
            	}
        	}
        }
        System.out.println(count);
    }
}

你可能感兴趣的:(算法)