牛客网--手串

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

思路

先分颜色记录每个颜色出现的位置；

然后再判断相邻位置是否小于m，如果小于则该颜色不符合；

同时对最末尾的位置进行单独判断，但是需要注意的是如果前面如果已经判断颜色不符合，这里可能还会再判断一次，所以需要避免重复计算

#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<stdio.h>
using namespace std;

int main() {
	int n, m, c;
	cin >> n >> m >> c;
	vector<vector<int>> num;
	for (int i = 0; i < c; i++) {
		vector<int> temp;
		num.push_back(temp);
	}

	//分颜色记录每个颜色出现的位置
	for (int i = 0; i < n; i++) {
		int num_i;
		cin >> num_i;
		int temp;
		for (int j = 0; j < num_i; j++) {
			cin >> temp;
			if (temp)
				num[temp - 1].push_back(i);
		}
	}

	int result = 0;
	for (int i = 0; i < c; i++) {
		int size = num[i].size();
		
		//防止同一种颜色重复计算
		int flag = 0;
		for (int j = 0; j < size- 1; j++) {
			if (num[i][j + 1] - num[i][j] + 1 <= m) {
				result++;
				flag = 1;
				break;
			}
		}

		if (!flag && num[i][0] + n- 1 - num[i][size - 1]+ 2 <= m) {
			result++;
		}
			
	}

	cout << result << endl;
	return 0;

}