今日头条2018校招后端第二批 - [编程题] 手串

版权声明:转载请注明出处 https://blog.csdn.net/yrwan95/article/details/81592669

时间限制:1秒

空间限制:65536K

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

输出例子1:

2

例子说明1:

第一种颜色出现在第1颗串珠,与规则无冲突。
第二种颜色分别出现在第 1,3,4颗串珠,第3颗与第4颗串珠相邻,所以不合要求。
第三种颜色分别出现在第1,3,5颗串珠,第5颗串珠的下一个是第1颗,所以不合要求。
总计有2种颜色的分布是有问题的。 
这里第2颗串珠是透明的。
package com.yrwan.java;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
import java.util.Map.Entry;

/*
 * 输入描述:
 * 第一行输入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)
 * 
 * 
 * 输出描述:
 * 一个非负整数,表示该手链上有多少种颜色不符需求。
 * 
 */
public class Toutiao2018_5 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();// 珠子数
		int m = sc.nextInt();// 间隔数最小
		int c = sc.nextInt();// 颜色数

		// 开始录入
		// Map<颜色的序号,在手链上的分布List>
		Map<Integer, ArrayList<Integer>> map = new HashMap<>();
		for (int i = 1; i <= c; i++) {
			map.put(i, new ArrayList<Integer>());
		}
		for (int i = 1; i <= n; i++) {// 第i个位置
			int num = sc.nextInt();// 有num种颜色
			for (int j = 0; j < num; j++) {// 对于第j种颜色
				int color = sc.nextInt();// 这种颜色的color号
				ArrayList<Integer> arrayList = map.get(color);// 获取color号的分布List
				arrayList.add(i);// 把第i位加上去
				map.put(color, arrayList);// 如此循环把位置信息录入到了Map里
			}
		}

		// 开始判断
		int err = 0;// 最初错误数为0
		for (int i = 1; i <= c; i++) {// 对于第i种颜色
			ArrayList<Integer> colorList = map.get(i);// 获取第i种颜色的分布List
			int[] colorArray = new int[colorList.size()];
			for (int j = 0; j < colorArray.length; j++) {// 第i种颜色的分布放在一个数组中
				colorArray[j] = colorList.get(j);
			}
			// 接下来在数组中判断是否符合要求
			for (int j = 0; j < colorArray.length; j++) {
				// 如果是非末位&&间距小于m
				if (j + 1 < colorArray.length && colorArray[j + 1] - colorArray[j] < m) {
					err++;
					break;
				}
				// 如果是末位&&与首位距离小于m
				if (j + 1 == colorArray.length && colorArray[0] + n - colorArray[j] < m) {
					err++;
					break;
				}
			}
		}
		System.out.println(err);
	}
}

 

阅读更多

没有更多推荐了,返回首页