今日头条笔试之手串问题

一、问题描述

题目摘自牛客网，可以到牛客网直接查找，现直接贴上原题目描述。

作为一个手串艺人，有金主向你订购了一条包含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个珠子中同时出现。因此最自然的思路就是遍历每一种颜色，判读该颜色的分布是否有问题。由于共有n个珠子，c种颜色，因此很自然的想到用一个nxc的矩阵存储每种颜色出现在哪个珠子上，初始化矩阵元素都为0，某种颜色出现在了某个珠子上，则其对应的元素置1，比如第一种颜色出现在了第2个珠子上，则a[1][1]=1（珠子下标从0开始，颜色下标从1开始）。因此按列遍历矩阵，遍历的时候记录每种颜色出现在了哪些珠子上，然后依次对这些珠子检查是否满足分布，有颜色不满足分布时，计数cnt++。最后要注意的是由于珠子是连成串的，因此判断时要对珠子下标取模n运算。

三、AC代码

#include <iostream>
#include <string.h>
using namespace std;

int main()
{
    int n,m,c,num,color,cnt=0;

    cin >> n >> m >> c;
    int b[n];
    int a[n][c+1];
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    for(int i=0;i<n;i++)
    {
        cin >> num;
        for(int j=0;j<num;j++)
        {
            cin >> color;
            a[i][color] = 1;
        }
    }

    for(int j=1;j<=c;j++)
    {
        int k=0;
        for(int i=0;i<n;i++)
        {
            if(a[i][j] == 1)
                b[k++] = i+1;
        }

        if(k != 1)
        {
            for(int i=0;i<k-1;i++)
            {
                if( b[i+1]-b[i] < m)
                {
                    cnt++;
                    break;
                }

            }

        }

    }
    cout << cnt <<endl;
    return 0;
}