算法101训练营-手串颜色问题
作为一个手串艺人,有金主向你订购了一条包含 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)
一个非负整数,表示该手链上有多少种颜色不符需求。
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 则不合格。
// 初始化串珠总个数,连续的串珠个数,颜色种类数,所有串珠的颜色信息数组, 同一颜色的串珠数组, 不合格的颜色个数
let ballNums,
linkNums,
colorNums,
ballColor = [],
sameColorBall = [],
count = 0;
[ballNums, linkNums, colorNums] = readline()
.split(' ')
.map((item) => Number(item));
// 依次保存每个串珠所用颜色信息
// 数组的每个元素都是一个数组,元素数组的第一位代表颜色个数,后续代表所用颜色
for (let i = 0; i < ballNums; ++i) {
ballColor[i] = readline()
.split(' ')
.map((item) => Number(item));
}
// 将同一颜色出现的串珠序号进行收集
ballColor.forEach((item, index) => {
// 若该串珠所用颜色种类大于0
// console.log(item, 'item');
if (item[0] > 0) {
let colorArr = item.slice(1);
// 下面的item代表不同的颜色种类
colorArr.forEach((color) => {
// 如果之前已经保存过使用某颜色的串珠序号,则直接将其添加到数组中去
if (sameColorBall[color]) {
sameColorBall[color].push(index + 1);
} else {
sameColorBall[color] = [index + 1];
}
});
}
});
sameColorBall.forEach((item) => {
// sameColorBall[0]代表使用‘0’这种颜色的所有串珠的序号数组,这里的序号是按顺序排列的
for (let i = 0; i < item.length - 1; ++i) {
if (item[i + 1] - item[i] < linkNums) {
++count;
break;
}
if (ballNums - item[item.length - 1] + item[0] < linkNums) {
++count;
break;
}
}
});
console.log(count);
缓一缓再解读…
先刷 MDN
评论0