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