题目描述
作为一个手串艺人,有金主向你订购了一条包含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>
#include <string.h>
int map[10005][55];
int result[55];
int temp[55];
int n,m,c,t;
int main(){
memset(map,0,10005*sizeof(map[55]));
memset(result,0,55*sizeof(int));
//依次输入
scanf("%d %d %d",&n,&m,&c);
for(int i=0;i<n;i++){
scanf("%d",&map[i][0]);
for(int j=1;j<=map[i][0];j++){
scanf("%d",&map[i][j]);
}
}
//珠子的数量
for(int i=0;i<n;i++){
memset(temp,0,55*sizeof(int));
//后面m-1个珠子
for(int j=0;j<m;j++){
//使用取余来模拟环状的珠子
int current = i+j;
if(current>=n)
current = current % n;
//遍历颜色,如果颜色存在,哈希表加一,大于一的出现次数不止一次
for(int k=1;k<=map[current][0];k++){
temp[map[current][k]]+=1;
if(temp[map[current][k]]>1)
result[map[current][k]]=1;
}
}
}
int sum = 0;
for(int i=1;i<=50;i++){
if(result[i]!=0){
sum+=1;
}
}
printf("%d\n",sum);
return 0;
}
上面这种方法比较容易想到,但是效率不高,等发现下面的说明后,发现竟然就是题目的做法,说明里说了,是看每种颜色出现的珠子,只要珠子的位置之差小于m就说明这个颜色不符合条件。
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,m,c;
cin >> n >> m >> c;
vector <int> map[c+1];
for(int i=0;i<n;i++){
int k;
cin >> k;
//根据颜色存储珠子的位置,以示例为例
//第一种颜色:1
//第二种颜色:1,3,4
for(int j=0;j<k;j++){
int t;
cin >> t;
map[t].push_back(i);
}
}
int count=0;
for(int i=1;i<=c;i++){
sort(map[i].begin(),map[i].end());
int size = map[i].size();
for(int j=1;j<size;j++){
//如果前后两个珠子间隔小于m则证明该颜色不符合条件
if(map[i][j]-map[i][j-1]<m){
count++;
break;
}
//如果当前的珠子加m大于n就取余,模拟环状,
//如果取余后的珠子位置大于该颜色出现的第一次珠子的位置,证明与第一次珠子重复
if((map[i][j]+m>=n)&&(map[i][j]+m)%n>map[i][0]){
count++;
break;
}
}
}
cout << count << endl;
}