跳到主要内容

[编程题]手串

[编程题]手串

字节跳动题目

链接:https://www.nowcoder.com/questionTerminal/429c2c5a984540d5ab7b6fa6f0aaa8b5?f=discussion 来源:牛客网

热度指数:4358时间限制:C/C++ 1秒,其他语言2秒空间限制:C/C++ 64M,其他语言128M

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

First Try

2020-10-07

以为是个很复杂的题目,其实只是简单的列表统计。最大的难点是不知道输入是怎么回事,刚接触牛客网还不了解套路。其次需要考虑到手串是一个环,最后一个和第一个珠子需要额外计算距离。另外m是表示连续m个珠子,而不是说之间的间隔为m。

anyway,这题目最难的就是读题了。

import collections

n, m, c = [int(i) for i in input().split()]
col = collections.defaultdict(list)
for i in range(n):
vals = [int(i) for i in input().split()]
for val in vals[1:]:
col[val].append(i)

errors = 0
for vals in col.values():
for idx, val in enumerate(vals):
if idx == 0:
continue
if val - vals[idx - 1] + 1<= m:
errors += 1
break
# 需要考虑到手串是一个环
if vals[0] + n - vals[-1] + 1 <= m:
errors += 1
break
print(errors)

Failed Try

while True:
try:
txt = input()
cmd.append([int(i) for i in txt.split(' ')])
if len(cmd) == 1:
cmdlen, m, c = cmd[0]
if len(cmd) == cmdlen + 1:
col = collections.defaultdict(list)
errors = 0
for idx, line in enumerate(cmd[1:]):
if len(line) == 1:
continue
for color in line[1:]:
col[color].append(idx)
for val in col.values():
for idx, i in enumerate(val):
if idx == 0:
continue
if i - val[idx - 1] <= m:
errors += 1
break
print(errors)
cmdlen = -1
cmd = []
except:
break