题解：2018校招真题（字节跳动：手串）

原题链接：https://www.nowcoder.com/practice/0bb1fad52f474bdaa4d7636ca3a98244?tpId=90&tqId=30819&tPage=9&rp=9&ru=/ta/2018test&qru=/ta/2018test/question-ranking

题目描述

作为一个手串艺人，有金主向你订购了一条包含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;
}