题目描述
作为一个手串艺人,有金主向你订购了一条包含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颗串珠是透明的。
思路:针对环形数组,我一般的做法是下标对长度取余,不仅代码简洁,也更容易理解,希望大家可以借鉴。对于该题,我们只需要保存每种颜色上次出现的柱子标号即可,之后判断当前位置和上一个位置的距离是否小于m,从而更新答案。
#include<stdio.h>
int n,m,c,a[10005][1005],last[55],flag[55];
int main(void){
scanf("%d%d%d",&n,&m,&c);
for(int i=0;i<n;i++){
scanf("%d",&a[i][0]);
for(int j=1;j<=a[i][0];j++)
scanf("%d",&a[i][j]);
}
int id=0,ans=0;
for(int i=1;i<=c;i++)
last[i]=-100000;
while(id<2*n){
for(int i=1;i<=a[id%n][0];i++){
if(id-last[a[id%n][i]]<m && flag[a[id%n][i]]==0){
flag[a[id%n][i]]=1;
ans++;
}
last[a[id%n][i]]=id;
}
id++;
}
printf("%d\n",ans);
return 0;
}